Searched defs:stride_h (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | maxpooling_op_gpu.cu.cc | 70 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 ()() 406 const int kernel_h, const int kernel_w, const int stride_h, in operator ()() 435 const int kernel_h, const int kernel_w, const int stride_h, in operator ()() 477 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() 140 const int kernel_w, const int stride_p, const int stride_h, in operator ()()
|
D | xsmm_conv2d_test.cc | 47 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
|
D | avgpooling_op_gpu.cu.cc | 88 const int kernel_w, const int stride_h, in RunAvePoolBackwardNHWC()
|
D | conv_grad_filter_ops.cc | 68 const int stride_h, const int stride_w, T* col_data) { in Im2col() 455 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DSlowBackpropFilterOp() local
|
D | mkl_conv_ops.h | 562 const int64 stride_h = GetTensorDim(strides_, data_format_, 'H'); in MklConvBackpropCommonOp() local
|
D | conv_grad_input_ops.cc | 68 const int stride_h, const int stride_w, T* im_data) { in Col2im() 592 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DSlowBackpropInputOp() local
|
D | conv_grad_ops_3d.cc | 69 const int pad_r, const int stride_p, const int stride_h, in Col2im() 118 const int pad_r, const int stride_p, const int stride_h, in Im2col()
|
D | conv_ops.cc | 306 const int64 stride_h = GetTensorDim(strides, data_format, 'H'); in InitConv2DParameters() local
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | transpose_conv_test.cc | 43 int stride_h) { in TransposeConvOpModel()
|