Home
last modified time | relevance | path

Searched refs:row_stride (Results 1 – 25 of 91) 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/python/kernel_tests/
Dpooling_ops_test.py1114 row_stride, argument
1158 strides = [1, 1, row_stride, col_stride]
1164 strides = [1, row_stride, col_stride, 1]
1192 row_stride, argument
1236 strides = [1, 1, row_stride, col_stride]
1240 strides = [1, row_stride, col_stride, 1]
1271 row_stride=1,
1285 row_stride=1,
1299 row_stride=1,
1313 row_stride=2,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h31 Filter filter, int row_stride, int col_stride,
41 input, filter, col_stride, row_stride, padding, col_dilation,
51 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
55 SpatialConvolutionFunc(d, output, input, filter, row_stride, col_stride, in operator()
60 typename TTypes<T, 4>::ConstTensor filter, int row_stride, in operator()
66 d, output, input, filter, row_stride, col_stride, row_dilation, in operator()
78 int row_stride, int col_stride, int row_dilation,
83 col_stride, row_stride, padding, col_dilation,
91 int row_stride, int col_stride, int row_dilation,
97 input.cast<float>(), filter.cast<float>(), col_stride, row_stride,
[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,
Davgpooling_op.h31 int window_cols, int row_stride, int col_stride, in operator()
37 row_stride, padding); in operator()
41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride, in operator()
Deigen_backward_spatial_convolutions.h103 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
153 0, ((outputRows - 1) * row_stride + kernelRowsEff - inputRows) / 2);
159 const TensorIndex padding_bottom = inputRows - (outputRows - 1) * row_stride -
287 col_in_stride, row_stride, col_stride, padding_top,
294 col_in_stride, row_stride, col_stride,
403 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
420 eigen_assert(!(row_stride > 1 && row_in_stride > 1));
468 0, (outputRows - 1) * row_stride + kernelRowsEff - inputRows);
476 const TensorIndex expanded_out_rows = (outputRows - 1) * row_stride + 1;
591 col_in_stride, 1, 1, row_stride,
[all …]
Dpooling_ops_common.h68 int row_stride; member
207 params.row_stride, params.col_stride, pt); in SpatialMaxPool()
242 const int32 row_stride = params.row_stride; in SpatialMaxPool() local
265 : (hpad - window_rows) / row_stride + 1; in SpatialMaxPool()
266 const int32 h_end = std::min(hpad / row_stride + 1, out_height); in SpatialMaxPool()
316 params.window_cols, params.row_stride, params.col_stride,
465 params.row_stride, params.col_stride, pt);
503 const int32 row_stride = params.row_stride;
526 : (hpad - window_rows) / row_stride + 1;
527 const int32 h_end = std::min(hpad / row_stride + 1, out_height);
[all …]
Davgpooling_op.cc204 params.row_stride, params.col_stride, pt); in Compute()
223 int window_cols, int row_stride, int col_stride, \
312 const int row_stride = stride_[1]; in Compute() local
325 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
336 in_rows, in_cols, window_rows, window_cols, row_stride, in Compute()
347 GetBroadcastSize(r, in_rows, window_rows, row_stride, in Compute()
572 const int row_stride = stride_[1]; in Compute() local
585 GetWindowedOutputSize(in_rows, window_rows, row_stride, in Compute()
600 row_stride, // stride_h in Compute()
Dconv_grad_input_ops.h113 int row_dilation, int col_dilation, int row_stride, in operator()
122 strides[input_h] = row_stride; in operator()
151 row_dilation, row_stride, padding, &expected_out_rows, &padding_top, in operator()
190 col_stride, row_stride, col_dilation, row_dilation); in operator()
197 col_stride, row_stride, col_dilation, row_dilation, padding_top, in operator()
208 int row_dilation, int col_dilation, int row_stride,
214 row_dilation, col_dilation, row_stride, col_stride, padding,
226 int input_rows, int input_cols, int row_stride,
239 int input_rows, int input_cols, int row_stride,
258 desc.u = row_stride;
Dconv_ops.cc81 const Tensor& filter, int row_stride, int col_stride, in operator ()()
87 if (filter.dim_size(0) == 1 && filter.dim_size(1) == 1 && row_stride == 1 && in operator ()()
127 input.tensor<T, 4>(), filter.tensor<T, 4>(), row_stride, col_stride, in operator ()()
135 input.tensor<T, 4>(), filter.tensor<T, 4>(), row_stride, col_stride, in operator ()()
147 int col_dilation, int row_stride, int col_stride, in operator ()()
173 LaunchGeneric<CPUDevice, T>()(ctx, input, filter, row_stride, col_stride, in operator ()()
184 int col_dilation, int row_stride, int col_stride, in operator ()()
212 ctx, input, filter, row_stride, col_stride, row_dilation, col_dilation, in operator ()()
649 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
679 row_dilation == 1 && col_dilation == 1 && row_stride == 1 && in operator ()()
[all …]
Dconv_ops.h38 int col_dilation, int row_stride, int col_stride,
49 int col_dilation, int row_stride, int col_stride,
/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.cc30 tensorflow::int64 output_cols, tensorflow::int64 row_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
39 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
52 tensorflow::int64 row_stride, tensorflow::int64 col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
60 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc33 tensorflow::int64 row_stride, tensorflow::int64 col_stride, in __xla_cpu_runtime_EigenConvF32() argument
44 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF32()
56 tensorflow::int64 output_cols, tensorflow::int64 row_stride, in __xla_cpu_runtime_EigenConvF16() argument
68 kernel_channels, kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc56 int64 output_rows, int64 output_cols, int64 row_stride, in MKLConvImpl() argument
75 memory::dims conv1_strides = {ToInt(row_stride), ToInt(col_stride)}; in MKLConvImpl()
159 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_MKLConvF32() argument
169 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConvF32()
175 kernel_filters, output_rows, output_cols, row_stride, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h32 tensorflow::int64 row_stride, tensorflow::int64 col_stride,
45 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
Druntime_conv2d.h31 tensorflow::int64 output_cols, tensorflow::int64 row_stride,
46 tensorflow::int64 row_stride, tensorflow::int64 col_stride,
/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/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h95 const DenseIndex row_stride = 1,
147 static_cast<float>(row_stride));
152 out_height = numext::ceil(InputRows / static_cast<float>(row_stride));
215 .extract_image_patches(kernelRows, kernelCols, row_stride, col_stride,
/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