tilelang.carver.roller.shape_inference.common¶

類別¶

`Statement`
`InputShapeInference`

Module Contents¶

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

參數:

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)¶

參數:: deps (list[Statement])

deps¶

infer(shape, rstep=None)¶

參數:: rstep (dict[str, int])

get_input_exprs(output_exprs)¶