Home
last modified time | relevance | path

Searched defs:stride_h (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.cu.cc71 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in MaxPoolForwardNCHW()
115 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
148 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC()
188 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolBackwardNoMaskNHWC()
277 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNCHW()
318 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNHWC()
402 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in operator ()()
421 const int kernel_h, const int kernel_w, const int stride_h, in operator ()()
452 const int kernel_h, const int kernel_w, const int stride_h, in operator ()()
497 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in operator ()()
Dpooling_ops_3d_gpu.cu.cc36 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNCDHW()
87 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNDHWC()
142 const int kernel_w, const int stride_p, const int stride_h, in operator ()()
Dxsmm_conv2d_test.cc47 int stride_h; member
131 int stride_h = param->stride_h; in naive_conv_fp() local
201 int stride_h = stride; in TEST_F() local
Davgpooling_op_gpu.cu.cc86 const int kernel_w, const int stride_h, in RunAvePoolBackwardNHWC()
Dconv_grad_input_ops_benchmark_test.cc41 int out_depth, int stride_h, int stride_w, in Conv2DBackpropInput()
Dconv_grad_filter_ops_benchmark_test.cc42 int out_depth, int stride_h, int stride_w, in Conv2DBackpropFilter()
Dconv_grad_filter_ops.cc73 const int stride_h, const int stride_w, T* col_data) { in Im2col()
272 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
Dconv_grad_input_ops.cc79 const int stride_h, const int stride_w, T* im_data) { in Col2im()
397 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DBackpropInputOp() local
Dconv_grad_ops_3d.cc77 const int pad_r, const int stride_p, const int stride_h, in Col2im()
126 const int pad_r, const int stride_p, const int stride_h, in Im2col()
Dconv_ops.cc372 const int64 stride_h = GetTensorDim(strides, data_format, 'H'); in InitConv2DParameters() local
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/tests/
Dtranspose_conv_test.cc28 int stride_w, int stride_h) { in QuantizedTransposeConvOpModel()
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc52 Padding padding, int stride_w, int stride_h, in BaseTransposeConvOpModel()
Dpooling_test.cc32 Padding padding = Padding_VALID, int stride_w = 2, int stride_h = 2, in BasePoolingOpModel()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h5490 const int stride_h, const int stride_w, T* im_data) { in Col2im()