tilelang.carver.roller.shape_inference.common¶

Classes¶

`Statement`
`InputShapeInference`

Module Contents¶

class tilelang.carver.roller.shape_inference.common.Statement(output, dependent_region, var_map, range_map)¶

Parameters:

output (str)
dependent_region (dict)
var_map (collections.OrderedDict)
range_map (collections.OrderedDict)

output¶

dependent_region¶

var_map¶

range_map¶

class tilelang.carver.roller.shape_inference.common.InputShapeInference(deps)¶

Parameters:: deps (List[Statement])

deps¶

infer(shape, rstep=None)¶

Parameters:: rstep (Dict[str, int])

get_input_exprs(output_exprs)¶