| /third_party/mindspore/tests/ut/python/model/ |
| D | resnet.py | 27 def _conv3x3(in_channel, out_channel, stride=1): argument
34 def _conv1x1(in_channel, out_channel, stride=1): argument
41 def _conv7x7(in_channel, out_channel, stride=1): argument
58 def _fc(in_channel, out_channel): argument
82 in_channel, argument
198 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src/ |
| D | resnet.py | 27 def _conv3x3(in_channel, out_channel, stride=1): argument
34 def _conv1x1(in_channel, out_channel, stride=1): argument
41 def _conv7x7(in_channel, out_channel, stride=1): argument
58 def _fc(in_channel, out_channel): argument
82 in_channel, argument
198 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | resnet.py | 27 def _conv3x3(in_channel, out_channel, stride=1): argument
34 def _conv1x1(in_channel, out_channel, stride=1): argument
41 def _conv7x7(in_channel, out_channel, stride=1): argument
58 def _fc(in_channel, out_channel): argument
82 in_channel, argument
198 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | resnet.py | 25 def _conv_variance_scaling_initializer(in_channel, out_channel, kernel_size): argument
40 def _conv3x3(in_channel, out_channel, stride=1, use_se=False): argument
50 def _conv1x1(in_channel, out_channel, stride=1, use_se=False): argument
60 def _conv7x7(in_channel, out_channel, stride=1, use_se=False): argument
80 def _fc(in_channel, out_channel, use_se=False): argument
110 in_channel, argument
275 …def _make_layer(self, block, layer_num, in_channel, out_channel, stride, use_se=False, se_block=Fa… argument
|
| /third_party/mindspore/tests/st/quantization/resnet50_quant/ |
| D | resnet_quant_manual.py | 36 def _conv3x3(in_channel, out_channel, stride=1): argument
43 def _conv1x1(in_channel, out_channel, stride=1): argument
50 def _conv7x7(in_channel, out_channel, stride=1): argument
67 def _fc(in_channel, out_channel): argument
122 in_channel, argument
244 def _make_layer(self, block, layer_num, in_channel, out_channel, stride): argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_fp16.cc | 32 int in_channel = filter_tensor->Channel(); in PackWeight() local
43 int in_channel = filter_tensor->Channel(); in MallocWeightBiasData() local
102 int in_channel = filter_tensor->Channel(); in Init() local
|
| D | convolution_winograd_fp16.cc | 37 int in_channel = weight_tensor->Channel(); in MallocWeightBiasData() local
153 int in_channel = weight_tensor->Channel(); in Init() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_fp32.cc | 83 size_t in_channel = filter_tensor->Channel(); in Init() local
172 int32_t in_channel = filter_tensor->Channel(); in PackWeight() local
203 size_t in_channel = filter_tensor->Channel(); in MallocWeightBiasData() local
|
| D | convolution_winograd_fp32.cc | 109 int in_channel = filter_tensor->Channel(); in Init() local
190 int in_channel = filter_tensor->Channel(); in MallocWeightBiasData() local
|
| D | adder_fp32.cc | 71 int in_channel = filter_tensor->Channel(); in InitWeightBias() local
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | conv2d_backprop_input_fusion.cc | 25 void Conv2DBackpropInputFusion::Init(int64_t in_channel, int64_t out_channel, const std::vector<int… in Init()
44 void Conv2DBackpropInputFusion::set_in_channel(int64_t in_channel) { in set_in_channel()
|
| D | conv2d_fusion.cc | 24 void Conv2DFusion::Init(int64_t in_channel, int64_t out_channel, const std::vector<int64_t> &kernel… in Init()
41 void Conv2DFusion::set_in_channel(const int64_t in_channel) { (void)this->AddAttr(kInChannel, MakeV… in set_in_channel()
|
| D | adder_fusion.cc | 22 void AdderFusion::Init(const int64_t in_channel, const int64_t out_channel, const std::vector<int64… in Init()
|
| D | conv2d_backprop_filter_fusion.cc | 49 void Conv2DBackpropFilterFusion::set_in_channel(const int64_t in_channel) { in set_in_channel()
|
| D | conv2d_transpose_fusion.cc | 22 void Conv2dTransposeFusion::Init(int64_t in_channel, int64_t out_channel, const std::vector<int64_t… in Init()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | update_conv2d_param_pass.cc | 25 … SetConvAttr(const PrimitivePtr &prim, const std::vector<int64_t> &kernel_size, int64_t in_channel, in SetConvAttr()
82 int64_t in_channel = format == mindspore::NHWC ? shape[kInputIndexThree] : shape[1]; in UpdateConv2DAttr() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_fp32_coder.cc | 32 int in_channel = conv_param_->input_channel_; in InitTmpBuffer() local
61 int in_channel = filter_tensor_->Channel(); in InitWeightBias() local
|
| D | deconv2d_fp32_coder.cc | 83 int in_channel = filter_tensor_->Channel(); in InitWeightBias() local
|
| /third_party/mindspore/tests/ut/python/pynative_mode/ge/ops/ |
| D | test_conv.py | 28 def get_me_conv_output(input_data, weight, in_channel, out_channel, kernel_size, argument
35 def __init__(self, weight, in_channel, out_channel, kernel_size, argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | adder.cc | 22 void Adder::Init(const int64_t in_channel, const int64_t out_channel, const std::vector<int64_t> &k… in Init()
36 void Adder::set_in_channel(const int64_t in_channel) { (void)this->AddAttr(kInChannel, MakeValue(in… in set_in_channel()
|
| D | conv2d_transpose.cc | 28 void Conv2DTranspose::Init(int64_t in_channel, int64_t out_channel, const std::vector<int64_t> &ker… in Init()
45 void Conv2DTranspose::set_in_channel(int64_t in_channel) { in set_in_channel()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | de_conv2d_grad_filter.cc | 24 void DeConv2DGradFilter::Init(const int64_t in_channel, const int64_t out_channel, in Init()
42 void DeConv2DGradFilter::set_in_channel(const int64_t in_channel) { in set_in_channel()
|
| D | group_conv2d_grad_input.cc | 23 void GroupConv2DGradInput::Init(const int64_t &in_channel, const int64_t &out_channel, in Init()
43 void GroupConv2DGradInput::set_in_channel(const int64_t &in_channel) { in set_in_channel()
|
| /third_party/mindspore/tests/st/networks/models/deeplabv3/src/backbone/ |
| D | resnet_deeplab.py | 23 def _conv_bn_relu(in_channel, argument
45 def _deep_conv_bn_relu(in_channel, argument
67 def _stob_deep_conv_btos_bn_relu(in_channel, argument
95 def _stob_conv_btos_bn_relu(in_channel, argument
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/ |
| D | pack_tests.cc | 66 int in_channel = conv_param->input_channel_; in TEST_F() local
129 int in_channel = conv_param->input_channel_; in TEST_F() local
|