Searched defs:replace_shape (Results 1 – 4 of 4) sorted by relevance
118 ValuePtr replace_shape = MakeValue(new_size_splits); in ReplaceNodeInputOrAttrs() local
176 ValuePtr replace_shape = MakeValue(replace_shape_); in ReplaceNodeInputOrAttrs() local
114 int64_t replace_shape = origin_ele / strategy[i - 1]; in ReplaceDynamicInput() local
285 int64_t replace_shape = origin_shape_ele / constant_shard_num; in ChangeDynamicDstShapeForSkipRedistribution() local312 int64_t replace_shape = origin_shape_ele / out_strategy[i - 1]; in ChangeDynamicDstShapeForSkipRedistribution() local