Home
last modified time | relevance | path

Searched refs:fuse_to_dynamic_shapes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dtensorexpr_fuser.cpp418 bool fuse_to_dynamic_shapes) in TensorExprFuser() argument
422 fuse_to_dynamic_shapes_(fuse_to_dynamic_shapes) { in TensorExprFuser()
1329 bool fuse_to_dynamic_shapes) { in FuseTensorExprs() argument
1339 fuse_to_dynamic_shapes, "Fusing static shapes with composed op NYI"); in FuseTensorExprs()
1346 graph, min_group_size, add_composed_op, fuse_to_dynamic_shapes); in FuseTensorExprs()
H A Dtensorexpr_fuser.h17 bool fuse_to_dynamic_shapes = false);