| /third_party/mindspore/tests/ut/python/communication/ |
| D | test_comm.py | 48 def __init__(self, input_channel, out_channel, op): argument
63 def __init__(self, input_channel, out_channel): argument
77 def __init__(self, input_channel, out_channel): argument
98 def __init__(self, input_channel, out_channel, op): argument
113 def __init__(self, input_channel, out_channel): argument
128 def __init__(self, batch_size, input_channel, out_channel): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_slidewindow_fp32.cc | 42 auto input_channel = filter_tensor->Channel(); in Init() local
190 auto input_channel = filter_tensor->Channel(); in PackWeight() local
203 auto input_channel = filter_tensor->Channel(); in MallocWeightBiasData() local
|
| D | deconvolution_fp32.cc | 58 auto input_channel = weight_tensor->Batch(); in MallocWeightBiasData() local
83 auto input_channel = weight_tensor->Batch(); in PackWeight() local
178 auto input_channel = weight_tensor->Batch(); in Init() local
|
| D | convolution_1x1_fp32.cc | 122 auto input_channel = filter_tensor->Channel(); in Init() local
273 auto input_channel = filter_tensor->Channel(); in PackWeight() local
300 auto input_channel = filter_tensor->Channel(); in MallocWeightBiasData() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_1x1_fp16.cc | 85 auto input_channel = weight_tensor->Channel(); in MallocWeightBiasData() local
116 auto input_channel = weight_tensor->Channel(); in PackWeight() local
153 auto input_channel = weight_tensor->Channel(); in Init() local
|
| D | deconvolution_fp16.cc | 58 auto input_channel = weight_tensor->Batch(); in PackWeight() local
70 auto input_channel = weight_tensor->Batch(); in MallocWeightBiasData() local
192 auto input_channel = weight_tensor->Batch(); in Init() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | conv2d_3x3_int8_coder.cc | 28 int input_channel = conv_param->input_channel_; in ProcessFilterUint8() local
48 int input_channel = conv_param_->input_channel_; in InitWeightBias() local
|
| D | conv2d_1x1_int8_coder.cc | 152 int32_t input_channel = filter_tensor_->Channel(); in InitWeightBias() local
|
| D | conv2d_int8_coder.cc | 87 int32_t input_channel = filter_tensor_->Channel(); in InitWeightBias() local
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_optimizer.py | 30 def __init__(self, input_channel, out_channel): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | convolution_3x3_int8.cc | 31 auto input_channel = conv_param->input_channel_; in ProcessFilterUint8() local
83 auto input_channel = filter_tensor->Channel(); in InitWeightBias() local
|
| D | convolution_1x1_int8.cc | 154 int Convolution1x1Int8CPUKernel::InitBiasByzp(const void *src_weight, int input_channel, int output… in InitBiasByzp()
216 auto input_channel = filter_tensor->Channel(); in InitWeightBias() local
262 auto input_channel = filter_tensor->Channel(); in InitWeightBiasArm32() local
|
| D | convolution_int8.cc | 52 auto input_channel = filter_tensor->Channel(); in InitWeightBias() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | depthwise_conv2d_infer.c | 39 int input_channel = input->shape_[3]; in DepthwiseConv2dInferShape() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/ |
| D | conv1x1_init_int8_wrapper.c | 21 int Conv1x1Init(int8_t *src_weight, int32_t *src_bias, int32_t *filter_zps, int32_t input_channel, in Conv1x1Init()
|
| D | conv_init_int8_wrapper.c | 23 … int kernel_w, int input_channel, int output_channel, int32_t input_zp, bool filter_peroc, in ConvInit()
|
| /third_party/mindspore/tests/st/fusion/ |
| D | test_conv_bn1_fusion.py | 25 input_channel = 2048 variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | space_to_depth_split.cc | 38 int64_t input_channel = SizeToLong(x_shape[kDim1]); in CreateTensor() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | deconv_fp32.c | 19 void PackDeConvWeightFp32(const float *weight, float *dst, int input_channel, int output_channel, i… in PackDeConvWeightFp32()
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | convolution_npu.cc | 181 auto input_channel = in_tensors.front().Shape()[NHWC_C]; in GetNPUConvOp() local
|
| /third_party/mindspore/tests/ut/python/onnx/ |
| D | test_onnx.py | 128 def __init__(self, input_channel, kernel_size): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | convolution_base.cc | 191 auto input_channel = filter_tensor->Channel(); in SetIfPerChannel() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/base/ |
| D | conv2d_base_coder.cc | 147 auto input_channel = static_cast<size_t>(filter_tensor_->Channel()); in SetIfPerChannel() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pack_int8.c | 21 … size_t plane_size, size_t input_channel, size_t output_channel) { in PackInputSum16x4PerChannelArm32()
52 size_t plane_size, size_t input_channel, size_t output_channel) { in PackInputSum16x4PerChannel()
81 …reOptPeroc(const int8_t *src_input, int8_t *packed_input, int32_t *input_sum, size_t input_channel, in Conv1x1PreOptPeroc()
509 …PreOptPert(const int8_t *src_input, int8_t *packed_input, int32_t *input_sum, size_t input_channel, in Conv1x1PreOptPert()
857 int input_channel = conv_param->input_channel_; in PackWeightToC8Int8() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | conv2d_transpose.cc | 332 int input_channel = conv_param->input_channel_; in OpenCLConv2dTransposeCreator() local
|