Home
last modified time | relevance | path

Searched defs:new_axis (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dtranspose_strategy.cc150 auto new_axis = kNH2NC[axis]; in ChangeCommonOp() local
179 auto new_axis = kNH2NC[axis]; in ChangeOpCrop() local
192 auto new_axis = kNC2NH[axis]; in ChangeOpCrop() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_transform.cc311 auto new_axis = axis; in OptimizeTensorRedistributionOperatorList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/
Dgrad_math_ops.cc219 auto new_axis = axis_; in Calc() local
2508 auto new_axis = inputs.at(1); in __anonbcb2e6c51602() local