Home
last modified time | relevance | path

Searched refs:row_stride (Results 1 – 25 of 95) sorted by relevance

1234

/external/gemmlowp/internal/
Doutput_neon.h397 int row_stride = dst->rows_stride();
405 vst1q_s32(dst_ptr + i * row_stride, transpose_top.buf.reg[i]);
414 vst1q_s32(dst_ptr + (i + 4) * row_stride, transpose_bottom.buf.reg[i]);
468 int row_stride = dst->rows_stride();
476 vst1q_s32(dst_ptr + i * row_stride, transpose_top_left.buf.reg[i]);
485 vst1q_s32(dst_ptr + (i + 4) * row_stride,
495 vst1q_s32(dst_ptr + i * row_stride + 4, transpose_top_right.buf.reg[i]);
504 vst1q_s32(dst_ptr + (i + 4) * row_stride + 4,
519 int row_stride = dst->rows_stride();
520 vst1q_lane_s32(dst_ptr + 0 * row_stride, src.buf.reg[0], 0);
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h41 Filter filter, int row_stride, int col_stride,
51 input, filter, col_stride, row_stride, padding, col_dilation,
64 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
68 SpatialConvolutionFunc(d, output, input, filter, row_stride, col_stride, in operator()
74 int row_stride, int col_stride, int row_dilation, in operator()
77 SpatialConvolutionFunc(d, output, input, filter, row_stride, col_stride, in operator()
83 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
89 d, output, input, filter, row_stride, col_stride, row_dilation, in operator()
96 int row_stride, int col_stride, int row_dilation, in operator()
101 d, output, input, filter, row_stride, col_stride, row_dilation, in operator()
[all …]
Dconv_grad_ops.h176 int row_dilation, int col_dilation, int row_stride,
186 int row_dilation, int col_dilation, int row_stride,
197 int col_dilation, int row_stride, int col_stride,
207 int row_dilation, int col_dilation, int row_stride,
Dmaxpooling_op.h33 int window_cols, int row_stride, int col_stride, in operator()
38 col_stride, row_stride, padding); in operator()
46 int window_cols, int row_stride, int col_stride,
Deigen_backward_spatial_convolutions.h99 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
149 0, ((outputRows - 1) * row_stride + kernelRowsEff - inputRows) / 2);
155 const TensorIndex padding_bottom = inputRows - (outputRows - 1) * row_stride -
268 col_in_stride, row_stride, col_stride, padding_top,
275 col_in_stride, row_stride, col_stride,
384 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
401 eigen_assert(!(row_stride > 1 && row_in_stride > 1));
449 0, (outputRows - 1) * row_stride + kernelRowsEff - inputRows);
457 const TensorIndex expanded_out_rows = (outputRows - 1) * row_stride + 1;
572 col_in_stride, 1, 1, row_stride,
[all …]
Dpooling_ops_common.h68 int row_stride; member
205 params.row_stride, params.col_stride, pt); in SpatialMaxPool()
240 const int32_t row_stride = params.row_stride; in SpatialMaxPool() local
264 : (hpad - window_rows) / row_stride + 1; in SpatialMaxPool()
265 const int32_t h_end = std::min(hpad / row_stride + 1, out_height); in SpatialMaxPool()
315 params.window_cols, params.row_stride, params.col_stride,
467 params.row_stride, params.col_stride, pt);
505 const int32_t row_stride = params.row_stride;
529 : (hpad - window_rows) / row_stride + 1;
530 const int32_t h_end = std::min(hpad / row_stride + 1, out_height);
[all …]
Dconv_ops.cc87 const Tensor& filter, int row_stride, int col_stride, in operator ()()
93 if (filter.dim_size(0) == 1 && filter.dim_size(1) == 1 && row_stride == 1 && in operator ()()
133 input.tensor<T, 4>(), filter.tensor<T, 4>(), row_stride, col_stride, in operator ()()
141 input.tensor<T, 4>(), filter.tensor<T, 4>(), row_stride, col_stride, in operator ()()
155 const Tensor& filter, int row_stride, int col_stride, in operator ()()
228 filter_slice, row_stride, col_stride, row_dilation, col_dilation, in operator ()()
236 filter_slice, row_stride, col_stride, row_dilation, col_dilation, in operator ()()
254 int col_dilation, int row_stride, int col_stride, in operator ()()
309 LaunchGrouped<T>()(ctx, input, filter, row_stride, col_stride, in operator ()()
313 LaunchGeneric<CPUDevice, T>()(ctx, input, filter, row_stride, col_stride, in operator ()()
[all …]
Davgpooling_op.cc215 params.row_stride, params.col_stride, pt); in Compute()
234 int window_cols, int row_stride, int col_stride, \
323 const int row_stride = stride_[1]; in Compute() local
336 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
347 in_rows, in_cols, window_rows, window_cols, row_stride, in Compute()
359 GetBroadcastSize(r, in_rows, window_rows, row_stride, in Compute()
584 const int row_stride = stride_[1]; in Compute() local
597 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
612 row_stride, // stride_h in Compute()
Davgpooling_op.h31 int window_cols, int row_stride, int col_stride, in operator()
38 row_stride, padding); in operator()
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dpooling_ops_test.py1156 row_stride, argument
1200 strides = [1, 1, row_stride, col_stride]
1206 strides = [1, row_stride, col_stride, 1]
1234 row_stride, argument
1278 strides = [1, 1, row_stride, col_stride]
1282 strides = [1, row_stride, col_stride, 1]
1313 row_stride=1,
1327 row_stride=1,
1341 row_stride=1,
1355 row_stride=2,
[all …]
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_state_sampler.c264 uint32_t row_stride[PIPE_MAX_TEXTURE_LEVELS]; in prepare_shader_sampling() local
302 row_stride[j] = lp_tex->row_stride[j]; in prepare_shader_sampling()
327 row_stride[0] = 0; in prepare_shader_sampling()
345 row_stride[0] = lp_tex->row_stride[0]; in prepare_shader_sampling()
357 row_stride, img_stride, mip_offsets); in prepare_shader_sampling()
417 uint32_t row_stride; in prepare_shader_images() local
458 row_stride = lp_img->row_stride[view->u.tex.level]; in prepare_shader_images()
467 row_stride = 0; in prepare_shader_images()
486 row_stride = lp_img->row_stride[0]; in prepare_shader_images()
496 row_stride, img_stride, in prepare_shader_images()
Dlp_texture.c124 lpr->row_stride[level] = nblocksx * block_size; in llvmpipe_texture_layout()
126 lpr->row_stride[level] = align(nblocksx * block_size, util_cpu_caps.cacheline); in llvmpipe_texture_layout()
129 if ((uint64_t)lpr->row_stride[level] * nblocksy > LP_MAX_TEXTURE_SIZE) { in llvmpipe_texture_layout()
134 lpr->img_stride[level] = lpr->row_stride[level] * nblocksy; in llvmpipe_texture_layout()
226 &lpr->row_stride[0] ); in llvmpipe_displaytarget_layout()
236 memset(map, 0, height * lpr->row_stride[0]); in llvmpipe_displaytarget_layout()
292 lpr->row_stride[0] = bytes; in llvmpipe_resource_create_all()
505 &lpr->row_stride[0]); in llvmpipe_resource_from_handle()
606 pt->stride = lpr->row_stride[level]; in llvmpipe_transfer_map_ms()
895 *stride = lpr->row_stride[0]; in llvmpipe_get_resource_info()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc27 int64_t output_rows, int64_t output_cols, int64_t row_stride, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16() argument
35 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
47 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32() argument
55 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
Druntime_conv2d.cc30 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_EigenConv2DF32() argument
41 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConv2DF32()
52 int64_t output_rows, int64_t output_cols, int64_t row_stride, in __xla_cpu_runtime_EigenConv2DF16() argument
63 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConv2DF16()
Druntime_conv2d_acl.cc34 int64_t output_cols, int64_t row_stride, int64_t col_stride, in ACLDepthwiseConvImpl() argument
54 col_stride, row_stride, static_cast<unsigned int>(padding_left), in ACLDepthwiseConvImpl()
133 int64_t row_stride, int64_t col_stride, in ACLGemmConvImpl() argument
157 col_stride, row_stride, static_cast<unsigned int>(padding_left), in ACLGemmConvImpl()
238 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_ACLConv2DF32() argument
253 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_ACLConv2DF32()
262 output_cols, row_stride, col_stride, padding_top, in __xla_cpu_runtime_ACLConv2DF32()
274 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_ACLConv2DF32()
Druntime_conv2d_mkl.cc57 int64_t output_rows, int64_t output_cols, int64_t row_stride, in MKLConvImpl() argument
77 memory::dims conv1_strides = {ToInt(row_stride), ToInt(col_stride)}; in MKLConvImpl()
159 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_MKLConv2DF32() argument
170 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConv2DF32()
176 kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_MKLConv2DF32()
Druntime_conv2d.h30 int64_t output_rows, int64_t output_cols, int64_t row_stride,
42 int64_t row_stride, int64_t col_stride, int64_t padding_top,
Druntime_single_threaded_conv2d.h31 int64_t row_stride, int64_t col_stride, int64_t padding_top,
42 int64_t output_rows, int64_t output_cols, int64_t row_stride,
/external/ComputeLibrary/src/core/NEON/kernels/convolution/common/
Dutils.cpp27 void PrintMatrix(const float* const m, const int M, const int N, const int row_stride) in PrintMatrix() argument
33 printf("%.3f ", m[i*row_stride + j]); in PrintMatrix()
/external/mesa3d/src/gallium/drivers/softpipe/
Dsp_state_sampler.c176 uint32_t row_stride[PIPE_MAX_TEXTURE_LEVELS]; in prepare_shader_sampling() local
215 row_stride[j] = sp_tex->stride[j]; in prepare_shader_sampling()
240 row_stride[0] = 0; in prepare_shader_sampling()
258 row_stride[0] = sp_tex->stride[0]; in prepare_shader_sampling()
269 row_stride, img_stride, mip_offsets); in prepare_shader_sampling()
/external/mesa3d/src/mesa/state_tracker/
Dst_draw_feedback.c315 uint32_t row_stride[PIPE_MAX_TEXTURE_LEVELS]; in st_feedback_draw_vbo() local
339 row_stride[j] = sv_transfer[i][j]->stride; in st_feedback_draw_vbo()
360 row_stride[0] = 0; in st_feedback_draw_vbo()
373 last_level, 0, 0, (void*)base_addr, row_stride, in st_feedback_draw_vbo()
391 unsigned width, height, num_layers, row_stride, img_stride; in st_feedback_draw_vbo() local
404 row_stride = img_transfer[i]->stride; in st_feedback_draw_vbo()
410 row_stride = 0; in st_feedback_draw_vbo()
420 num_layers, addr, row_stride, img_stride, 0, 0); in st_feedback_draw_vbo()
/external/mesa3d/src/mesa/drivers/dri/nouveau/
Dnouveau_texture.c422 const GLvoid *src, GLvoid *dst, int row_stride) in nouveau_compressed_copy() argument
438 dst += row_stride; in nouveau_compressed_copy()
474 int row_stride; in nouveau_teximage() local
480 &map, &row_stride); in nouveau_teximage()
485 pixels, map, row_stride); in nouveau_teximage()
489 row_stride, in nouveau_teximage()
567 int row_stride; in nouveau_texsubimage() local
571 GL_MAP_WRITE_BIT, &map, &row_stride); in nouveau_texsubimage()
576 pixels, map, row_stride); in nouveau_texsubimage()
580 row_stride, &map, in nouveau_texsubimage()
/external/libaom/aom_dsp/
Dbutteraugli.c88 uint32_t row_stride; in aom_calc_butteraugli() local
89 JxlButteraugliResultGetDistmap(result, &distmap, &row_stride); in aom_calc_butteraugli()
100 dist_map[j * width + i] = distmap[j * row_stride + i]; in aom_calc_butteraugli()
/external/mesa3d/src/gallium/drivers/panfrost/
Dpan_sfbd.c95 fb->color_writeback.row_stride = stride; in panfrost_sfbd_set_cbuf()
101 fb->color_writeback.row_stride *= 16; in panfrost_sfbd_set_cbuf()
123 fb->zs_writeback.row_stride = rsrc->slices[level].stride * 16; in panfrost_sfbd_set_zsbuf()
230 params.crc_buffer.row_stride = slice->checksum_stride; in panfrost_sfbd_fragment()
/external/webrtc/test/testsupport/
Djpeg_frame_writer.cc73 int row_stride = input_frame.width() * kColorPlanes; in WriteFrame() local
75 row_pointer[0] = &rgb_buf.get()[cinfo.next_scanline * row_stride]; in WriteFrame()

1234