Home
last modified time | relevance | path

Searched refs:conv1x1 (Results 1 – 16 of 16) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dconv_1x1_int8_tests.cc127 kernel::Convolution1x1Int8CPUKernel *conv1x1 = in TEST_F() local
130 conv1x1->Init(); in TEST_F()
131 conv1x1->Run(); in TEST_F()
134 delete conv1x1; in TEST_F()
196 kernel::Convolution1x1Int8CPUKernel *conv1x1 = in TEST_F() local
199 conv1x1->Init(); in TEST_F()
200 conv1x1->Run(); in TEST_F()
203 delete conv1x1; in TEST_F()
273 auto *conv1x1 = in TEST_F() local
276 conv1x1->Init(); in TEST_F()
[all …]
/third_party/mindspore/tests/perf_test/
Dresnet_example.py30 def conv1x1(in_channels, out_channels, stride=1, padding=0, pad_mode='pad'): function
50 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
56 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
62 self.conv_down_sample = conv1x1(in_channels, out_channels,
/third_party/mindspore/tests/st/tbe_networks/
Dresnet.py39 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
87 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
93 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
130 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
136 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
142 self.conv_down_sample = conv1x1(in_channels, out_channels, stride=stride, padding=0)
/third_party/mindspore/tests/ut/python/transform/
Dtest_transform.py40 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
62 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
68 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
75 self.conv_down_sample = conv1x1(in_channels, out_channels,
/third_party/mindspore/tests/st/networks/models/
Dresnetv1_5.py45 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
102 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
108 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
145 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
151 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
157 self.conv_down_sample = conv1x1(in_channels, out_channels, stride=stride, padding=0)
/third_party/mindspore/tests/st/mem_reuse/
Dresnet.py50 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
106 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
112 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
149 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
155 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
161 self.conv_down_sample = conv1x1(in_channels, out_channels, stride=stride, padding=0)
/third_party/mindspore/tests/st/export_and_load/
Dtest_resnet_air.py63 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
121 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
127 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
164 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
170 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
176 self.conv_down_sample = conv1x1(
/third_party/mindspore/tests/ut/python/model/
Dres18_example.py33 def conv1x1(in_channels, out_channels, stride=1, padding=0, pad_mode='pad'): function
53 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
59 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
65 self.conv_down_sample = conv1x1(in_channels, out_channels,
/third_party/mindspore/tests/ut/python/exec/
Dresnet_example.py33 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
67 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
73 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
79 self.conv_down_sample = conv1x1(in_channels, out_channels,
/third_party/mindspore/tests/st/networks/
Dtest_gpu_resnet.py69 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
127 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
133 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
170 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
176 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
182 self.conv_down_sample = conv1x1(
/third_party/mindspore/tests/ut/python/parallel/
Dtest_operator_model_parallel.py115 def conv1x1(in_channels, out_channels, stride=1): function
191 self.conv1 = conv1x1(in_channels, out_chls, stride=1)
197 self.conv3 = conv1x1(out_chls, out_channels, stride=1)
236 self.conv1 = conv1x1(in_channels, out_chls, stride=1)
242 self.conv3 = conv1x1(out_chls, out_channels, stride=1)
250 self.conv_down_sample = conv1x1(in_channels, out_channels, stride=stride)
/third_party/mindspore/tests/st/pynative/
Dtest_pynative_resnet50_gpu.py62 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
110 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
116 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
153 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
159 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
165 self.conv_down_sample = conv1x1(in_channels, out_channels, stride=stride, padding=0)
Dtest_pynative_resnet50_ascend.py100 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
148 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
154 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
191 self.conv1 = conv1x1(in_channels, out_chls, stride=stride, padding=0)
197 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
203 self.conv_down_sample = conv1x1(in_channels, out_channels, stride=stride, padding=0)
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_1x1_fp32.cc166 auto conv1x1 = reinterpret_cast<Convolution1x1CPUKernel *>(cdata); in Convolution1x1Run() local
167 auto error_code = conv1x1->DoConv1x1(task_id); in Convolution1x1Run()
213 auto conv1x1 = reinterpret_cast<Convolution1x1CPUKernel *>(cdata); in Convolution1x1RunHw() local
214 auto error_code = conv1x1->DoConv1x1Hw(task_id); in Convolution1x1RunHw()
/third_party/mindspore/tests/ut/python/communication/
Dtest_data_parallel_resnet.py35 def conv1x1(in_channels, out_channels, stride=1, padding=0, pad_mode='pad'): function
55 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
61 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
67 self.conv_down_sample = conv1x1(in_channels, out_channels,
/third_party/mindspore/tests/ut/python/ops/
Dtest_nn_ops.py46 def conv1x1(in_channels, out_channels, stride=1, padding=0): function
70 self.conv1 = conv1x1(in_channels, out_chls, stride=1, padding=0)
76 self.conv3 = conv1x1(out_chls, out_channels, stride=1, padding=0)
82 self.conv_down_sample = conv1x1(in_channels, out_channels,