Home
last modified time | relevance | path

Searched defs:shape_map (Results 1 – 25 of 25) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Darray_len.cc53 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in ArrayLenInferInner() local
91 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in InferValue() local
Dsize.cc59 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_shape_ptr); in InferValue() local
Dimag.cc50 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShap… in ImagInferShape() local
Dchannel_shuffle.cc47 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in ChannelShuffleInferShape() local
Dadaptive_max_pool2d.cc61 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in AdaptiveMaxPool2DInferShape() local
Dpolar.cc47 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in PolarInferShape() local
Dget_squeeze_slice_shape.cc51 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(data_abs->GetShape()); in GetSqueezeSliceShapeInferInner() local
Dlinear_sum_assignment.cc50 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShap… in LinearSumAssignmentInferShape() local
Ddata_format_dim_map.cc49 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in DataFormatDimMapInferShape() local
Dtril.cc39 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_shape->GetShape()); in TrilInferShape() local
Drandperm.cc103 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(shape_ptr); in InferShape() local
Dnormalize_tuple_index.cc220 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(data_abs->GetShape()); in NormalizeTupleIndexInferInner() local
255 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in InferShape() local
Dtopk.cc39 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShap… in TopKInferShape() local
Ddepth_to_space.cc77 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShap… in DepthToSpaceInferShape() local
Drandom_gamma.cc66 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShap… in GammaInferShape() local
Dstrided_slice_v2.cc318 ShapeMap shape_map; in DynamicComputeInferShapeV2() local
411 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[x_index]->GetShape()); in StridedSliceV2InferShape() local
Dmax_pool.cc133 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in MaxPoolInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dmax_pool_grad.cc43 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in MaxPoolGradInferShape() local
Dslice_grad.cc34 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->GetShape()); in SliceGradInferShape() local
Dsoftmax_grad.cc51 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->GetShape()); in SoftmaxGradInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dsort_ext.cc42 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShap… in InferShape() local
Dtopk_ext.cc44 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->GetShap… in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/utils/
Dcheck_convert_utils.cc664 ShapeMap shape_map; in ConvertShapePtrToShapeMap() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/
Dtensor_index.cc50 ShapeMap shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(data->BuildShape()); in PreHandleIndex() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_ge/
Dge_graph_executor.cc1068 std::map<std::string, std::string> shape_map; in UpdateInputShapeOption() local