Home
last modified time | relevance | path

Searched refs:SPLIT_BY_AXIS (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dredistribution_operator_infer.cc111 if (InsertOperator(SPLIT_BY_AXIS, args) == Status::FAILED) { in InferSplitByAxis()
152 if (InsertOperator(SPLIT_BY_AXIS, args_allsplit) == Status::FAILED) { in InferPermuteByAxis()
196 if (name == SPLIT_BY_AXIS) { in InsertOperator()
/third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/
Dredistribution_operator_infer_test.cc58 if (str == SPLIT_BY_AXIS) { in InferOperatorCheck()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dops_utils.h237 constexpr char SPLIT_BY_AXIS[] = "SplitByAxis"; variable