Home
last modified time | relevance | path

Searched defs:stride_w (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.cu.cc70 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNCHW()
112 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
144 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC()
183 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolBackwardNoMaskNHWC()
270 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW()
310 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNHWC()
389 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in operator ()()
407 const int stride_w, const int pad_t, const int pad_l, T* top_data, in operator ()()
436 const int stride_w, const int pad_t, const int pad_l, const T* top_diff, in operator ()()
477 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in operator ()()
Dpooling_ops_3d_gpu.cu.cc35 const int stride_p, const int stride_h, const int stride_w, const int pad_p, in MaxPoolGradBackwardNoMaskNCDHW()
85 const int stride_p, const int stride_h, const int stride_w, const int pad_p, in MaxPoolGradBackwardNoMaskNDHWC()
141 const int stride_w, const int pad_p, const int pad_t, const int pad_l, in operator ()()
Dxsmm_conv2d_test.cc48 int stride_w; member
132 int stride_w = param->stride_w; in naive_conv_fp() local
200 int stride_w = stride; in TEST_F() local
Davgpooling_op_gpu.cu.cc89 const int stride_w, const int pad_t, in RunAvePoolBackwardNHWC()
Dconv_grad_filter_ops.cc68 const int stride_h, const int stride_w, T* col_data) { in Im2col()
456 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DSlowBackpropFilterOp() local
Ddepthwise_conv_op.cc278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeOp() local
Dmkl_conv_ops.h563 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in MklConvBackpropCommonOp() local
Dconv_grad_input_ops.cc68 const int stride_h, const int stride_w, T* im_data) { in Col2im()
593 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DSlowBackpropInputOp() local
Ddepthwise_conv_grad_op.cc552 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropInputOp() local
1026 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropFilterOp() local
Dconv_grad_ops_3d.cc70 const int stride_w, T* im_data) { in Col2im()
119 const int stride_w, T* col_data) { in Im2col()
Dconv_ops.cc307 const int64 stride_w = GetTensorDim(strides, data_format, 'W'); in InitConv2DParameters() local
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc42 const TensorData& output, Padding padding, int stride_w, in TransposeConvOpModel()