Home
last modified time | relevance | path

Searched refs:input_stride (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconcat_base.c38 int input_stride = after_axis_size * inputs_output_shape[i][axis]; in Concat() local
40 int offset = UP_DIV(input_stride, thread_num); in Concat()
41 int count = input_stride - offset * task_id; in Concat()
48 const uint8_t *src = src_base + j * input_stride + task_id * offset; in Concat()
Dsplit_with_over_lap_base.c27 int input_stride = param->split_dim_size_ * param->inner_stride_ * param->element_bytes_; in DoSplitWithOverlapParallel() local
35 src_ptr += input_stride; in DoSplitWithOverlapParallel()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_depthwise_fp32.h60 int output_width, size_t input_stride, size_t relu, size_t relu6);
63 int output_width, size_t input_stride, size_t relu, size_t relu6);
111 size_t output_width, size_t input_stride, size_t relu, size_t relu6);
114 size_t output_width, size_t input_stride, size_t relu, size_t relu6);
130 int output_width, int input_stride, bool relu, bool relu6, int kernel);
Dconv_depthwise_fp32.c770 int output_width, int input_stride, bool relu, bool relu6, int kernel) { in ConvDwFp32IndirectRow() argument
801 input = input + input_stride; in ConvDwFp32IndirectRow()
808 int output_width, int input_stride, bool relu, bool relu6, int kernel) { in ConvDwFp32IndirectRow() argument
810 …ConvDwFp32Indirect3x3(output, input, weights, bias, channels, output_width, input_stride * sizeof(… in ConvDwFp32IndirectRow()
813 …ConvDwFp32Indirect5x5(output, input, weights, bias, channels, output_width, input_stride * sizeof(… in ConvDwFp32IndirectRow()
821 int output_width, int input_stride, bool relu, bool relu6, int kernel) { in ConvDwFp32IndirectRow() argument
823 …ConvDwFp32Avx3x3(output, input, weights, bias, channels, output_width, input_stride * sizeof(float… in ConvDwFp32IndirectRow()
825 …ConvDwFp32Avx5x5(output, input, weights, bias, channels, output_width, input_stride * sizeof(float… in ConvDwFp32IndirectRow()
838 int input_stride = conv_param->kernel_h_ * step_w; in ConvDwIndirection() local
855 conv_param->output_w_, input_stride, relu, relu6, 9); in ConvDwIndirection()
[all …]
Dpack_fp32.c50 int input_stride = (input_h * in_w + input_w) * in_channel; in Im2ColPackUnitFp32() local
57 int input_y_stride = j * in_w * in_channel + input_stride; in Im2ColPackUnitFp32()
65 int input_y_stride = j * dilation_h * in_w * in_channel + input_stride; in Im2ColPackUnitFp32()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/
DConvDwFp32IndirectRow.c28 size_t output_width, size_t input_stride, size_t relu, size_t relu6) { in ConvDwFp32Avx5x5() argument
29 input_stride /= sizeof(float *); in ConvDwFp32Avx5x5()
37 input += input_stride; in ConvDwFp32Avx5x5()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dpooling_int8.c111 int input_stride = (in_h_index * in_w + in_w_index) * channel; in AvgPoolingOptInt8() local
139 int in_offset = in_channel_offset + input_stride + (h * in_w + w) * channel; in AvgPoolingOptInt8()
211 int in_offset = in_channel_offset + input_stride + (h * in_w + w) * channel; in AvgPoolingOptInt8()
261 int in_offset = in_channel_offset + input_stride + (h * in_w + w) * channel; in AvgPoolingOptInt8()
450 int input_stride = (in_h_index * in_w + in_w_index) * channel + in_batch_offset; in MaxPoolingOptInt8() local
460 int in_h_offset = input_stride + h * in_w * channel + c; in MaxPoolingOptInt8()
Dpack_int8.c763 int input_stride = input_h * in_w * in_channel + input_w * in_channel; in Im2ColPackUnitInt8Opt() local
770 int input_y_stride = j * in_w * in_channel + input_stride; in Im2ColPackUnitInt8Opt()
777 int input_y_stride = j * dilation_h * in_w * in_channel + input_stride; in Im2ColPackUnitInt8Opt()
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_vs_exec.c98 unsigned input_stride, in vs_exec_run_linear() argument
166 input = (const float (*)[4])((const char *)input + input_stride); in vs_exec_run_linear()
Ddraw_tess.c167 unsigned input_stride = input_verts->vertex_size; in draw_tess_ctrl_shader_run() local
176 shader->input_vertex_stride = input_stride; in draw_tess_ctrl_shader_run()
333 unsigned input_stride = input_verts->vertex_size; in draw_tess_eval_shader_run() local
351 shader->input_vertex_stride = input_stride; in draw_tess_eval_shader_run()
Ddraw_vs.h142 unsigned input_stride,
Ddraw_vs_llvm.c57 unsigned input_stride, in vs_llvm_run_linear() argument
Ddraw_gs.c561 unsigned input_stride = input_verts->vertex_size; in draw_geometry_shader_run() local
623 shader->input_vertex_stride = input_stride; in draw_geometry_shader_run()
/third_party/mesa3d/src/amd/vulkan/
Dradv_query.c154 nir_ssa_def *input_stride = nir_imm_int(&b, db_count * 16); in build_occlusion_query_shader() local
155 nir_ssa_def *input_base = nir_imul(&b, input_stride, global_id); in build_occlusion_query_shader()
289 nir_ssa_def *input_stride = nir_imm_int(&b, pipelinestat_block_size * 2); in build_pipeline_statistics_query_shader() local
290 nir_ssa_def *input_base = nir_imul(&b, input_stride, global_id); in build_pipeline_statistics_query_shader()
435 nir_ssa_def *input_stride = nir_imm_int(&b, 32); in build_tfb_query_shader() local
436 nir_ssa_def *input_base = nir_imul(&b, input_stride, global_id); in build_tfb_query_shader()
559 nir_ssa_def *input_stride = nir_imm_int(&b, 8); in build_timestamp_query_shader() local
560 nir_ssa_def *input_base = nir_imul(&b, input_stride, global_id); in build_timestamp_query_shader()
/third_party/mesa3d/src/gallium/auxiliary/translate/
Dtranslate_generic.c63 unsigned input_stride; member
621 (ptrdiff_t)tg->attrib[attr].input_stride * index; in generic_run_one()
634 tg->attrib[attr].input_stride, in generic_run_one()
742 tg->attrib[i].input_stride = stride; in generic_set_buffer()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpack_fp16.c61 int input_stride = (input_h * in_w + input_w) * in_channel; in Im2ColPackUnitFp16() local
68 int input_y_stride = j * in_w * in_channel + input_stride; in Im2ColPackUnitFp16()
76 int input_y_stride = j * dilation_h * in_w * in_channel + input_stride; in Im2ColPackUnitFp16()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Ddepthwise_conv2d_info.cc42 int64_t input_stride = split_dim_size * inner_stride * element_bytes; in SplitConstantData() local
48 src_ptr += input_stride; in SplitConstantData()