Home
last modified time | relevance | path

Searched defs:col_stride (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc30 int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
48 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc32 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_EigenConvF32()
50 int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc57 int64 col_stride, int64 padding_top, int64 padding_bottom, in MKLConvImpl()
157 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_impl.h33 int64 row_stride, int64 col_stride, int64 padding_top, in EigenConvImpl()
Dir_emitter.cc1160 int64 col_stride = in HandleConvolution() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h57 Filter filter, int row_stride, int col_stride, in SpatialConvolutionFunc()
Dpooling_ops_common.h61 int col_stride; member
205 const int32 col_stride = params.col_stride; in SpatialMaxPool() local
Dpooling_ops_3d.h63 int col_stride; member
Davgpooling_op.cc276 const int col_stride = stride_[2]; in Compute() local
516 const int col_stride = stride_[2]; in Compute() local
Dconv_ops_fused_impl.h248 LaunchFusedConv2DWithOutputKernel(int row_stride, int col_stride, // in LaunchFusedConv2DWithOutputKernel()
Dmaxpooling_op.cc111 const int32 col_stride = params.col_stride; in SpatialMaxPoolWithArgMaxHelper() local
591 const int32 col_stride = params.col_stride; in SpatialMaxPoolGradGrad() local
Dpooling_ops_3d.cc590 const int32 col_stride = params.col_stride; in launch() local
Dconv_grad_filter_ops.cc546 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()()
Dops_util_test.cc38 int col_stride; member
Dconv_grad_input_ops.cc683 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()()
Dconv_ops.cc581 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()()
Dmkl_pooling_ops_common.h379 int col_stride; member
Deigen_spatial_convolutions_test.cc255 const int col_stride = 2; in TEST() local
635 int col_stride = 2; in TEST() local
/external/mesa3d/src/mesa/drivers/dri/radeon/
Dradeon_maos_vbtmp.h47 GLuint tc0_stride, tc1_stride, col_stride, spec_stride, fog_stride; in TAG() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py951 col_stride, argument
1027 col_stride, argument
1252 window_cols, row_stride, col_stride, padding, v2): argument
1276 window_rows, window_cols, row_stride, col_stride, argument
1703 window_cols, row_stride, col_stride, padding): argument
/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py161 filter_width, row_stride, col_stride, padding_type): argument
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc443 const Tensor& filter_param, int32 row_stride, int32 col_stride, in launch()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h699 const Index row_stride, const Index col_stride, in extract_image_patches()
/external/mesa3d/src/compiler/spirv/
Dvtn_variables.c822 unsigned num_ops, vec_width, col_stride; in _vtn_block_load_store() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc3283 int col_stride = 1; in TEST_F() member