Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/core/ops/
Dshape.cc36 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in ShapeInfer() local
57 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in ShapeInferValue() local
Dlog1p.cc31 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in Log1pInferShape() local
Dsoftplus.cc31 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in SoftplusInferShape() local
Dgelu.cc35 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
Dexp.cc37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
Derfinv.cc36 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in ErfinvInferShape() local
Dreciprocal.cc37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
Dlog.cc37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
Dneg.cc37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
Dlog_softmax.cc41 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in LogSoftmaxInferShape() local
Dsoftmax.cc53 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in SoftMaxInferShape() local
Done_hot.cc34 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in OneHotInferShape() local
Dsquare.cc37 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildSh… in SquareInferShape() local
Dreluv2.cc63 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
Dtile.cc60 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in TileInferShape() local
Dbias_add.cc40 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
/third_party/mindspore/mindspore/core/ops/grad/
Dbias_add_grad.cc48 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.cc410 ShapeMap shape_map; in ConvertShapePtrToShapeMap() local