Searched refs:CheckPositiveVector (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | conv2d_backprop_input.cc | 195 …(void)AddAttr(kKernelSize, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel… in set_kernel_size()
199 …(void)AddAttr(kStride, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kStride, stride, name()… in set_stride()
203 …(void)AddAttr(kDilation, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kDilation, dilation, … in set_dilation()
222 (void)AddAttr(kPad, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kPad, pad, name()))); in set_pad()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | max_pool.cc | 37 … MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel_size, this->name()))); in set_kernel_size()
42 …(void)this->AddAttr(kStrides, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kStrides, stride… in set_strides()
|
| D | avg_pool.cc | 37 … MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel_size, this->name()))); in set_kernel_size()
42 …(void)this->AddAttr(kStrides, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kStrides, stride… in set_strides()
|
| D | conv2d.cc | 270 …(void)AddAttr(kKernelSize, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel… in set_kernel_size()
274 …(void)AddAttr(kStride, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kStride, stride, name()… in set_stride()
278 …(void)AddAttr(kDilation, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kDilation, dilation, … in set_dilation()
297 (void)AddAttr(kPad, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kPad, pad, name()))); in set_pad()
|
| D | layer_norm_beta_gamma_backprop_v2.cc | 51 …(void)AddAttr(kShapeGamma, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kShapeGamma, shape_… in set_shape_gamma()
|
| D | zeros.cc | 35 (void)CheckAndConvertUtils::CheckPositiveVector("shape", out_shape, prim_name); in ZerosInferShape()
|
| D | ones.cc | 34 (void)CheckAndConvertUtils::CheckPositiveVector("shape", out_shape, prim_name); in OnesInferShape()
|
| D | conv2d_transpose.cc | 96 (void)AddAttr(kPad, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kPad, pad, name()))); in set_pad()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | avg_pool_fusion.cc | 75 (void)CheckAndConvertUtils::CheckPositiveVector(kStride, strides, op_name); in InferShape()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 155 …static std::vector<int64_t> CheckPositiveVector(const std::string &arg_name, const std::vector<int…
|
| D | check_convert_utils.cc | 309 std::vector<int64_t> CheckAndConvertUtils::CheckPositiveVector(const std::string &arg_name, in CheckPositiveVector() function in mindspore::CheckAndConvertUtils
|