tilelang.carver.roller.shape_inference.common module#
- class tilelang.carver.roller.shape_inference.common.InputShapeInference(deps: List[Statement])#
Bases:
object- get_input_exprs(output_exprs)#
- infer(shape, rstep: Optional[Dict[str, int]] = None)#
- class tilelang.carver.roller.shape_inference.common.Statement(output: str, dependent_region: dict, var_map: OrderedDict, range_map: OrderedDict)#
Bases:
object