| /third_party/mindspore/mindspore/core/ops/ |
| D | log_softmax.cc | 41 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in LogSoftmaxInferShape() local
42 if (shape_map.empty()) { in LogSoftmaxInferShape()
46 auto in_shape = shape_map[kShape]; in LogSoftmaxInferShape()
47 auto min_shape = shape_map[kMinShape]; in LogSoftmaxInferShape()
48 auto max_shape = shape_map[kMaxShape]; in LogSoftmaxInferShape()
|
| D | log1p.cc | 31 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in Log1pInferShape() local
32 auto in_shape = shape_map[kShape]; in Log1pInferShape()
33 auto min_shape = shape_map[kMinShape]; in Log1pInferShape()
34 auto max_shape = shape_map[kMaxShape]; in Log1pInferShape()
|
| D | softplus.cc | 31 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in SoftplusInferShape() local
32 auto in_shape = shape_map[kShape]; in SoftplusInferShape()
33 auto min_shape = shape_map[kMinShape]; in SoftplusInferShape()
34 auto max_shape = shape_map[kMaxShape]; in SoftplusInferShape()
|
| D | softmax.cc | 53 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in SoftMaxInferShape() local
54 if (shape_map.empty()) { in SoftMaxInferShape()
58 auto in_shape = shape_map[kShape]; in SoftMaxInferShape()
59 auto min_shape = shape_map[kMinShape]; in SoftMaxInferShape()
60 auto max_shape = shape_map[kMaxShape]; in SoftMaxInferShape()
|
| D | gelu.cc | 35 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
36 auto in_shape = shape_map[kShape]; in InferShape()
37 auto min_shape = shape_map[kMinShape]; in InferShape()
38 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| D | shape.cc | 36 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in ShapeInfer() local
37 auto in_shape = shape_map[kShape]; in ShapeInfer()
57 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in ShapeInferValue() local
58 auto inshape = shape_map[kShape]; in ShapeInferValue()
|
| D | log.cc | 37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
38 auto in_shape = shape_map[kShape]; in InferShape()
39 auto min_shape = shape_map[kMinShape]; in InferShape()
40 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| D | exp.cc | 37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
38 auto in_shape = shape_map[kShape]; in InferShape()
39 auto min_shape = shape_map[kMinShape]; in InferShape()
40 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| D | neg.cc | 37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
38 auto in_shape = shape_map[kShape]; in InferShape()
39 auto min_shape = shape_map[kMinShape]; in InferShape()
40 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| D | reciprocal.cc | 37 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
38 auto in_shape = shape_map[kShape]; in InferShape()
39 auto min_shape = shape_map[kMinShape]; in InferShape()
40 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| D | tile.cc | 60 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in TileInferShape() local
61 auto input_shape = shape_map[kShape]; in TileInferShape()
62 auto min_shape = shape_map[kMinShape]; in TileInferShape()
63 auto max_shape = shape_map[kMaxShape]; in TileInferShape()
|
| D | one_hot.cc | 34 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in OneHotInferShape() local
35 auto in_shape = shape_map[kShape]; in OneHotInferShape()
36 auto min_shape = shape_map[kMinShape]; in OneHotInferShape()
37 auto max_shape = shape_map[kMaxShape]; in OneHotInferShape()
|
| D | square.cc | 37 …auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->BuildSh… in SquareInferShape() local
38 auto in_shape = shape_map[kShape]; in SquareInferShape()
39 auto min_shape = shape_map[kMinShape]; in SquareInferShape()
40 auto max_shape = shape_map[kMaxShape]; in SquareInferShape()
|
| D | reluv2.cc | 63 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
64 auto input_shape = shape_map[kShape]; in InferShape()
65 auto min_shape = shape_map[kMinShape]; in InferShape()
66 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| D | bias_add.cc | 40 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
41 auto input_shape = shape_map[kShape]; in InferShape()
42 auto min_shape = shape_map[kMinShape]; in InferShape()
43 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| D | erfinv.cc | 36 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in ErfinvInferShape() local
37 auto in_shape = shape_map[kShape]; in ErfinvInferShape()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | bias_add_grad.cc | 48 auto shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape()); in InferShape() local
49 auto input_shape = shape_map[kShape]; in InferShape()
50 auto min_shape = shape_map[kMinShape]; in InferShape()
51 auto max_shape = shape_map[kMaxShape]; in InferShape()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.cc | 410 ShapeMap shape_map; in ConvertShapePtrToShapeMap() local
411 shape_map[kShape] = shape_element->shape(); in ConvertShapePtrToShapeMap()
412 shape_map[kMinShape] = shape_element->min_shape(); in ConvertShapePtrToShapeMap()
413 shape_map[kMaxShape] = shape_element->max_shape(); in ConvertShapePtrToShapeMap()
414 return shape_map; in ConvertShapePtrToShapeMap()
|