Searched refs:GetTensorInputShape (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | layer_norm_x_backprop_v2.cc | 31 auto x_shape = CheckAndConvertUtils::GetTensorInputShape(primitive->name(), input_args, 0); in LayerNormXBackpropV2InferShape()
32 …auto res_for_gamma_shape = CheckAndConvertUtils::GetTensorInputShape(primitive->name(), input_args… in LayerNormXBackpropV2InferShape()
|
| D | dropout_do_mask.cc | 45 auto x_shape = CheckAndConvertUtils::GetTensorInputShape(op_name, input_args, 0); in InferShape()
46 auto mask_shape = CheckAndConvertUtils::GetTensorInputShape(op_name, input_args, 1); in InferShape()
|
| D | ones_like.cc | 33 return CheckAndConvertUtils::GetTensorInputShape(op_name, input_args, 0); in InferShape()
|
| D | zeros_like.cc | 33 return CheckAndConvertUtils::GetTensorInputShape(op_name, input_args, 0); in InferShape()
|
| D | LayerNormXBackprop.cc | 31 auto x_shape = CheckAndConvertUtils::GetTensorInputShape(primitive->name(), input_args, 1); in LayerNormXBackpropInferShape()
|
| D | reduce_sum.cc | 85 auto shape_ptr = CheckAndConvertUtils::GetTensorInputShape("ReduceSum", input_args, 0); in InferShape()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | conv2d_backprop_filter.cc | 35 auto x_shape = CheckAndConvertUtils::GetTensorInputShape(prim_name, input_args, kInputIndex); in TransStrideTo4D()
36 auto dout_shape = CheckAndConvertUtils::GetTensorInputShape(prim_name, input_args, kDoutIndex); in TransStrideTo4D()
68 …auto shape_ptr = CheckAndConvertUtils::GetTensorInputShape(prim_name, input_args, kFilterSizeIdex); in Conv2DBackpropFilterInferShape()
|
| D | dropout_grad.cc | 55 auto shape = CheckAndConvertUtils::GetTensorInputShape(op_name, input_args, dy_index); in DropoutGradInfer()
|
| D | conv2d_backprop_input.cc | 97 auto shape_ptr = CheckAndConvertUtils::GetTensorInputShape(prim_name, input_args, kSizeIndex); in Conv2DBackpropInputInferShape()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 221 static abstract::ShapePtr GetTensorInputShape(const std::string &prim_name,
|
| D | check_convert_utils.cc | 417 abstract::ShapePtr CheckAndConvertUtils::GetTensorInputShape(const std::string &prim_name, in GetTensorInputShape() function in mindspore::CheckAndConvertUtils
|