Searched refs:CheckAttrIntOrTupleInt (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | conv2d_backprop_input.cc | 38 …CheckAndConvertUtils::CheckAttrIntOrTupleInt("kernel_size", primitive->GetAttr(kKernelSize), prim_… in SetPadList()
39 …auto stride = CheckAndConvertUtils::CheckAttrIntOrTupleInt("stride", primitive->GetAttr(kStride), … in SetPadList()
40 …auto dilation = CheckAndConvertUtils::CheckAttrIntOrTupleInt("dilation", primitive->GetAttr(kDilat… in SetPadList()
131 … out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("input x size", input_size_v, prim_name); in Conv2DBackpropInputInferShape()
|
| D | conv2d_backprop_filter.cc | 102 … out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("filter_size", filter_size_v, prim_name); in Conv2DBackpropFilterInferShape()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | zeros.cc | 34 …std::vector<int64_t> out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("shape", shape_value… in ZerosInferShape()
|
| D | ones.cc | 33 …std::vector<int64_t> out_shape = CheckAndConvertUtils::CheckAttrIntOrTupleInt("shape", shape_value… in OnesInferShape()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 307 …static std::vector<int64_t> CheckAttrIntOrTupleInt(const std::string &prim_name, const ValuePtr &a…
|
| D | check_convert_utils.cc | 686 std::vector<int64_t> CheckAndConvertUtils::CheckAttrIntOrTupleInt(const std::string &arg_name, cons… in CheckAttrIntOrTupleInt() function in mindspore::CheckAndConvertUtils
|