Searched defs:stride_w (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | maxpooling_op_gpu.cu.cc | 69 const int stride_w, const int pad_t, in MaxPoolForwardNCHW() 111 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_w, const int pad_t, in MaxPoolForwardNHWC() 183 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolBackwardNoMaskNHWC() 267 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW() 307 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNHWC() 383 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in operator ()() 401 const int stride_w, const int pad_t, const int pad_l, T* top_data, in operator ()() 428 const int stride_w, const int pad_t, const int pad_l, const T* top_diff, in operator ()() 465 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in operator ()()
|
D | pooling_ops_3d_gpu.cu.cc | 35 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 ()()
|
D | xsmm_conv2d_test.cc | 48 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
|
D | avgpooling_op_gpu.cu.cc | 88 const int stride_w, const int pad_t, in RunAvePoolBackwardNHWC()
|
D | depthwise_conv_op.cc | 273 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeOp() local
|
D | conv_grad_filter_ops.cc | 62 const int stride_h, const int stride_w, T* col_data) { in Im2col() 549 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DSlowBackpropFilterOp() local
|
D | conv_grad_input_ops.cc | 61 const int stride_h, const int stride_w, T* im_data) { in Col2im() 624 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DSlowBackpropInputOp() local
|
D | depthwise_conv_grad_op.cc | 538 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropInputOp() local 914 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropFilterOp() local
|
D | conv_ops.cc | 278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DOp() local
|