Home
last modified time | relevance | path

Searched refs:GetAndCheckFormat (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/core/ops/grad/
Dbias_add_grad.cc47 int64_t format = CheckAndConvertUtils::GetAndCheckFormat(primitive->GetAttr("format")); in InferShape()
Dconv2d_backprop_input.cc138 auto format = CheckAndConvertUtils::GetAndCheckFormat(primitive->GetAttr(kFormat)); in Conv2DBackpropInputInferShape()
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc32 int64_t GetAndCheckFormat(const ValuePtr &value) { in GetAndCheckFormat() function
153 int64_t data_format = GetAndCheckFormat(data_format_ptr); in InferImplBatchNorm()
267 int64_t data_format = GetAndCheckFormat(primitive->GetAttr("format")); in InferImplConv2D()
362 data_format = GetAndCheckFormat(data_format_ptr); in InferImplBiasAdd()
/third_party/mindspore/mindspore/core/ops/
Dbias_add.cc55 data_format = CheckAndConvertUtils::GetAndCheckFormat(data_format_ptr); in InferShape()
Dconv2d.cc175 int64_t data_format = CheckAndConvertUtils::GetAndCheckFormat(primitive->GetAttr("format")); in Conv2dInferShape()
/third_party/mindspore/mindspore/core/utils/
Dcheck_convert_utils.h312 static int64_t GetAndCheckFormat(const ValuePtr &value);
Dcheck_convert_utils.cc741 int64_t CheckAndConvertUtils::GetAndCheckFormat(const ValuePtr &value) { in GetAndCheckFormat() function in mindspore::CheckAndConvertUtils