| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | depth_to_space_infer.c | 44 …if (block_size == 0 || input_shape[kNHWC_C] % (block_size * block_size) != 0 || input_shape[kNHWC_… in DepthToSpaceInferShape()
52 output_shape[kNHWC_C] = input_shape[kNHWC_C] / (block_size * block_size); in DepthToSpaceInferShape()
|
| D | space_to_depth_infer.c | 53 if (input->shape_[kNHWC_C] == 0 || block_size * block_size > INT_MAX / input->shape_[kNHWC_C]) { in SpaceToDepthInferShape()
56 outputs[0]->shape_[kNHWC_C] = input->shape_[kNHWC_C] * (block_size * block_size); in SpaceToDepthInferShape()
|
| D | batch_to_space_infer.c | 60 output_shape[kNHWC_C] = input_shape[kNHWC_C]; in SetOutputShapeFromParam()
108 output_shape[kNHWC_C] = input_shape[kNHWC_C]; in SetOutputShapeFromInput()
|
| D | space_to_batch_nd_infer.c | 58 outputs[0]->shape_[kNHWC_C] = input->shape_[kNHWC_C]; in SpaceSetOutputShapeFromParam()
101 output_shape[kNHWC_C] = input->shape_[kNHWC_C]; in SpaceSetOutputShapeFromInput()
|
| D | space_to_batch_infer.c | 59 outputs[0]->shape_[kNHWC_C] = input->shape_[kNHWC_C]; in SpaceToBatchInferShape()
|
| /third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
| D | skip_gram_infer_test.cc | 32 inputs[0]->format_ = kNHWC_C; in TEST_F()
39 ASSERT_EQ(outputs[0]->format_, kNHWC_C); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | mul_int8.cc | 75 bool condition4 = in_tensor0_shape[kNHWC_C] == in_tensor1_shape[kNHWC_C]; in CheckSameShapeSize()
93 if (in_tensor0->ElementsNum() == in_tensor1->shape()[kNHWC_C]) { in CheckIfFastImpl()
97 if (in_tensor1->ElementsNum() == in_tensor0->shape()[kNHWC_C]) { in CheckIfFastImpl()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | pooling_fp16_grad.cc | 65 pool_param->input_channel_ = in_shape[kNHWC_C]; in ReSize()
69 pool_param->output_channel_ = out_shape[kNHWC_C]; in ReSize()
|
| D | convolution_fp16_grad_filter.cc | 50 conv_param->input_channel_ = x_tensor->shape().at(kNHWC_C); in ReSize()
51 conv_param->output_channel_ = dy_tensor->shape().at(kNHWC_C); in ReSize()
|
| D | bn_fp16_grad.cc | 60 int channels = input_x->shape().at(kNHWC_C); in ReSize()
|
| D | convolution_fp16_grad_input.cc | 48 conv_param->input_channel_ = dx_tensor->shape()[(kNHWC_C)]; in ReSize()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | pooling_grad.cc | 65 pool_param->input_channel_ = in_shape[kNHWC_C]; in ReSize()
69 pool_param->output_channel_ = out_shape[kNHWC_C]; in ReSize()
|
| D | deconvolution_grad_filter.cc | 53 conv_param->input_channel_ = x_tensor->shape().at(kNHWC_C); in Init()
54 conv_param->output_channel_ = dy_tensor->shape().at(kNHWC_C); in Init()
|
| D | convolution_grad_filter.cc | 53 conv_param->input_channel_ = x_tensor->shape().at(kNHWC_C); in ReSize()
54 conv_param->output_channel_ = dy_tensor->shape().at(kNHWC_C); in ReSize()
|
| D | bn_grad.cc | 47 int channels = input_x->shape().at(kNHWC_C); in ReSize()
|
| D | convolution_grad_input.cc | 48 conv_param->input_channel_ = dx_tensor->shape()[(kNHWC_C)]; in ReSize()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | space_to_batch_nd.cc | 49 param->padded_in_shape_[kNHWC_C] = in_tensors_[0]->shape().at(kNHWC_C); in CheckSpecs()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_conv_parser.cc | 145 if (INT_MUL_OVERFLOW(data_shape[kNHWC_C], multiplier)) { in Parse()
149 prim->set_out_channel(data_shape[kNHWC_C] * multiplier); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_conv_parser.cc | 84 if (INT_MUL_OVERFLOW_THRESHOLD(dims.at(kNHWC_C), group, INT64_MAX)) { in GetConvChannel()
88 *channel_in = dims.at(kNHWC_C) * group; in GetConvChannel()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | deconvolution_depthwise_fp16.cc | 39 conv_param_->input_channel_ = out_tensors_.front()->shape().at(kNHWC_C); in InitSlideParam()
43 conv_param_->output_channel_ = in_tensors_.front()->shape().at(kNHWC_C); in InitSlideParam()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | deconvolution_depthwise_fp32.cc | 38 conv_param_->input_channel_ = out_tensors_.front()->shape().at(kNHWC_C); in InitSlideParam()
42 conv_param_->output_channel_ = in_tensors_.front()->shape().at(kNHWC_C); in InitSlideParam()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/cmsis-nn/int8/ |
| D | conv2d_int8_coder.cc | 99 MS_CHECK_TRUE(input_tensor_->Channel() == filter_tensor_->DimensionSize(kNHWC_C), in SetParameters()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_utils.cc | 307 if (nhwc_axis > kNHWC_C) { in ConvertAxisFromNHWC2NCHW()
317 case kNHWC_C: in ConvertAxisFromNHWC2NCHW()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | resize_fp32_coder.cc | 177 int c = input_tensor_->shape().at(kNHWC_C); in DoCode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | op_base.h | 97 #define kNHWC_C 3 macro
|