Home
last modified time | relevance | path

Searched defs:shape_base (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dconcat_offset.cc66 ShapeVector shape_base = tensor_base->GetShape()->GetShapeVector(); in InferShape() local
Ddropout_gen_mask.cc138 auto shape_base = shape_args->GetShape(); in DropoutGenMaskInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/abstract/
Dparam_validator.cc123 auto shape_base = tensor_base->GetShape(); in CheckShapeSame() local
171 ShapePtr shape_base = tensor_base->shape(); in CheckShapeSame() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dscatter_nd.cc97 auto shape_base = input_args[kInputIndex2]->GetShape(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dneighbor_exchange_v2_unify_mindir.cc828 auto shape_base = shape_info.second; in CreatePadNode() local