Home
last modified time | relevance | path

Searched refs:out_c (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dwinograd_utils.h32 int dst_step, int out_c, int r_w, int r_h, int r_c);
168 MS_STQ_F32(dst_data + out_c, m[1]); \
169 MS_STQ_F32(dst_data + dst_step * out_c, m[2]); \
170 MS_STQ_F32(dst_data + dst_step * out_c + out_c, m[3]);
174 MS_STQ_F32(dst_data + out_c, m[1]); \
175 MS_STQ_F32(dst_data + 2 * out_c, m[2]); \
176 MS_STQ_F32(dst_data + dst_step * out_c, m[3]); \
177 MS_STQ_F32(dst_data + dst_step * out_c + out_c, m[4]); \
178 MS_STQ_F32(dst_data + dst_step * out_c + 2 * out_c, m[5]); \
179 MS_STQ_F32(dst_data + 2 * dst_step * out_c, m[6]); \
[all …]
Dwinograd_avx.h27 int dst_step, int out_c, int r_w, int r_h, int r_c);
153 MS_ST256_F32(dst_data + out_c, m[1]); \
154 MS_ST256_F32(dst_data + dst_step * out_c, m[2]); \
155 MS_ST256_F32(dst_data + dst_step * out_c + out_c, m[3]);
159 MS_ST256_F32(dst_data + out_c, m[1]); \
160 MS_ST256_F32(dst_data + 2 * out_c, m[2]); \
161 MS_ST256_F32(dst_data + dst_step * out_c, m[3]); \
162 MS_ST256_F32(dst_data + dst_step * out_c + out_c, m[4]); \
163 MS_ST256_F32(dst_data + dst_step * out_c + 2 * out_c, m[5]); \
164 MS_ST256_F32(dst_data + 2 * dst_step * out_c, m[6]); \
[all …]
Dwinograd_avx.c21 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2AvxUnit() argument
42 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2AvxUnit()
45 dst_data[i + dst_k_offset + k * out_c] = MS_F32X8_GETI(m[k + m_k_offset], i); in OutputTransform4x2AvxUnit()
53 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2ReluAvxUnit() argument
77 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2ReluAvxUnit()
80 dst_data[i + dst_k_offset + k * out_c] = MS_F32X8_GETI(m[k + m_k_offset], i); in OutputTransform4x2ReluAvxUnit()
88 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2Relu6AvxUnit() argument
115 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2Relu6AvxUnit()
118 dst_data[i + dst_k_offset + k * out_c] = MS_F32X8_GETI(m[k + m_k_offset], i); in OutputTransform4x2Relu6AvxUnit()
126 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x3AvxUnit() argument
[all …]
Dwinograd_utils.c328 int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2Unit() argument
350 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2Unit()
353 dst_data[i + dst_k_offset + k * out_c] = MS_F32X4_GETI(m[k + m_k_offset], i); in OutputTransform4x2Unit()
379 int dst_k_offset = k * dst_step * out_c; in OutputTransform4x2Unit()
382 dst_data[i + dst_k_offset + j * out_c] = m[j + m_k_offset] + bias_data[i]; in OutputTransform4x2Unit()
390 int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2ReluUnit() argument
415 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2ReluUnit()
418 dst_data[i + dst_k_offset + k * out_c] = MS_F32X4_GETI(m[k + m_k_offset], i); in OutputTransform4x2ReluUnit()
444 int dst_k_offset = k * dst_step * out_c; in OutputTransform4x2ReluUnit()
449 dst_data[i + dst_k_offset + j * out_c] = out_value; in OutputTransform4x2ReluUnit()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_utils_fp16.h33 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c);
291 vst1q_f16(dst_data + out_c, m[1]); \
292 vst1q_f16(dst_data + dst_step * out_c, m[2]); \
293 vst1q_f16(dst_data + dst_step * out_c + out_c, m[3]);
297 vst1_f16(dst_data + out_c, m[1]); \
298 vst1_f16(dst_data + dst_step * out_c, m[2]); \
299 vst1_f16(dst_data + dst_step * out_c + out_c, m[3]);
303 vst1q_f16(dst_data + out_c, m[1]); \
304 vst1q_f16(dst_data + 2 * out_c, m[2]); \
305 vst1q_f16(dst_data + dst_step * out_c, m[3]); \
[all …]
Dwinograd_utils_fp16.c463 int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2UnitFp16() argument
485 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2UnitFp16()
488 vst1q_f16(dst_data + dst_k_offset + k * out_c, m[k + m_k_offset]); in OutputTransform4x2UnitFp16()
514 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2UnitFp16()
517 dst_data[i + dst_k_offset + k * out_c] = m[k + m_k_offset][i]; in OutputTransform4x2UnitFp16()
543 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2UnitFp16()
546 dst_data[z + dst_k_offset + k * out_c] = m[k + m_k_offset]; in OutputTransform4x2UnitFp16()
553 … int src_step, int dst_step, int out_c, int r_w, int r_h, int r_c) { in OutputTransform4x2ReluUnitFp16() argument
578 int dst_k_offset = j * dst_step * out_c; in OutputTransform4x2ReluUnitFp16()
581 vst1q_f16(dst_data + dst_k_offset + k * out_c, m[k + m_k_offset]); in OutputTransform4x2ReluUnitFp16()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/
DConvDwFp32Row_sse.c29 size_t out_c = 0; in ConvDwFp32Row() local
30 for (; out_c < out_c16; out_c += C16NUM) { in ConvDwFp32Row()
55 for (; out_c < out_c8; out_c += C8NUM) { in ConvDwFp32Row()
70 for (; out_c < out_c4; out_c += C4NUM) { in ConvDwFp32Row()
80 for (; out_c < output_channel; out_c++) { in ConvDwFp32Row()
81 *output_ptr++ += weight_ptr[out_c] * input_ptr[out_c]; in ConvDwFp32Row()
/third_party/mesa3d/src/vulkan/util/
Dvk_extensions_gen.py211 extensions, out_c, out_h, includes = []): argument
233 if out_c:
234 with open(out_c, 'w') as f:
254 extensions, args.out_c, args.out_h, [])
Dvk_entrypoints_gen.py222 assert os.path.dirname(args.out_c) == os.path.dirname(args.out_h)
243 with open(args.out_c, 'w') as f:
Dvk_commands_gen.py128 with open(args.out_c, 'wb') as f:
Dvk_cmd_queue_gen.py406 assert os.path.dirname(args.out_c) == os.path.dirname(args.out_h)
431 with open(args.out_c, 'wb') as f:
Dvk_dispatch_table_gen.py952 if args.out_c:
953 with open(args.out_c, 'w') as f:
Dvk_physical_device_features.py235 with open(args.out_c, 'wb') as f:
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.c51 int out_c = conv_param->output_channel_; in SelectOutputUnit() local
62 float common_cost = (float)out_h * out_w * in_c * out_c * kernel_h * kernel_w; in SelectOutputUnit()
70 …float wino_cost = ((2 + out_c) * (float)input_unit * input_unit * in_c + ((float)input_unit + i) *… in SelectOutputUnit()
/third_party/unity/auto/
Dcolour_prompt.rb99 def out_c(mode, colour, str) method in ColourCommandLine
114 ColourCommandLine.new.out_c(:puts, role, str)
118 ColourCommandLine.new.out_c(:print, role, str)
/third_party/cJSON/tests/unity/auto/
Dcolour_prompt.rb98 def out_c(mode, colour, str) method in ColourCommandLine
113 ColourCommandLine.new.out_c(:puts, role, str)
117 ColourCommandLine.new.out_c(:print, role, str)
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dhuffman_encode.cc174 unsigned char out_c; in DoHuffmanCompress() local
202 out_c = 0; in DoHuffmanCompress()
205 out_c += tmp_c << ((quant::kMaxBit - 1) - (i % quant::kMaxBit)); in DoHuffmanCompress()
207 encode_str[2] += out_c; in DoHuffmanCompress()
208 out_c = 0; in DoHuffmanCompress()
/third_party/mindspore/tests/st/heterogeneous_excutor/
Dtest_heterogeneous_excutor.py27 def __init__(self, in_c, out_c): argument
35 self.bn2 = nn.BatchNorm2d(num_features=out_c,
41 out_channels=out_c,
/third_party/gstreamer/gstplugins_base/gst-libs/gst/audio/
Daudio-channel-mixer.c382 gint out_c[3] = { -1, -1, -1 }; in gst_audio_channel_mixer_fill_others() local
401 out_c, &out_has_center, out_r, &out_has_rear, in gst_audio_channel_mixer_fill_others()
416 gst_audio_channel_mixer_fill_one_other (matrix, in_f, out_c, in gst_audio_channel_mixer_fill_others()
419 gst_audio_channel_mixer_fill_one_other (matrix, in_s, out_c, in gst_audio_channel_mixer_fill_others()
423 gst_audio_channel_mixer_fill_one_other (matrix, in_r, out_c, in gst_audio_channel_mixer_fill_others()
455 in_f, out_c, RATIO_CENTER_FRONT); in gst_audio_channel_mixer_fill_others()
460 gst_audio_channel_mixer_fill_one_other (matrix, in_f, out_c, in gst_audio_channel_mixer_fill_others()
499 gst_audio_channel_mixer_fill_one_other (matrix, in_s, out_c, in gst_audio_channel_mixer_fill_others()
522 gst_audio_channel_mixer_fill_one_other (matrix, in_r, out_c, in gst_audio_channel_mixer_fill_others()
529 gst_audio_channel_mixer_fill_one_other (matrix, in_b, out_c, in gst_audio_channel_mixer_fill_others()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.cc34 int64_t out_c = weight_shape.at(kShapeN); in ApproximateFLOPs() local
43 return (input_h / stride_h) * (input_w / stride_w) * in_c * k_h * k_w * out_c / kPerFlops; in ApproximateFLOPs()
/third_party/mindspore/tests/st/mix_precision/
Dtest_mix_precision.py43 def __init__(self, in_c, out_c): argument
51 self.bn2 = nn.BatchNorm2d(num_features=out_c,
57 out_channels=out_c,
/third_party/mindspore/tests/st/ops/gpu/
Dtest_conv3d_op.py115 out_c = 2
150 net = MSConv3dNet(x.shape[1], out_c, kernel_size, weight_init=w)
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dresize_int8.cc336 auto out_c = out_tensor->Channel(); in RunImpl() local
342 auto out_ptr = output_data + start_index * out_c; in RunImpl()
346 … out_tensor->Height(), out_tensor->Width(), out_c, start_index, count, resize_quant_arg_); in RunImpl()
349 … input->Width(), out_tensor->Height(), out_tensor->Width(), out_c, in RunImpl()
/third_party/gstreamer/gstplugins_good/gst/deinterlace/tvtime/
Dgreedyh.c198 guint8 out_l, out_c; in greedyh_scanline_C_yuy2() local
300 out_c = CLAMP (best_c, min_c, max_c); in greedyh_scanline_C_yuy2()
318 Dest[1] = out_c; in greedyh_scanline_C_yuy2()
342 guint8 out_l, out_c; in greedyh_scanline_C_uyvy() local
444 out_c = CLAMP (best_c, min_c, max_c); in greedyh_scanline_C_uyvy()
462 Dest[0] = out_c; in greedyh_scanline_C_uyvy()