Home
last modified time | relevance | path

Searched defs:axis_value_ptr (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/view/
Dunstack_strides_calc.cc35 auto axis_value_ptr = prim->GetAttr(kAxis); in UnstackCalc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dtranspose_fusion.cc305 auto axis_value_ptr = prim->GetAttr(ops::kAxis); in AdjustAxis() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dconstruct_operator.cc117 ValuePtr axis_value_ptr = MakeValue(std::make_shared<Int64Imm>(axis)); in CreateSplitOp() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc480const ValueSequencePtr &axis_value_ptr, const PrimitivePtr &primitive) { in DoInferReduceShape()
629 auto axis_value_ptr = axis_value->cast<ValueSequencePtr>(); in InferImplReduceShape() local