Home
last modified time | relevance | path

Searched refs:pooled_height (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.cu.cc68 const int height, const int width, const int pooled_height, in MaxPoolForwardNCHW() argument
74 int ph = (index / pooled_width) % pooled_height; in MaxPoolForwardNCHW()
75 int c = (index / pooled_width / pooled_height) % channels; in MaxPoolForwardNCHW()
76 int n = index / pooled_width / pooled_height / channels; in MaxPoolForwardNCHW()
110 const int width, const int channels, const int pooled_height, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
118 int ph = (index / pooled_width) % pooled_height; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
119 int c = (index / pooled_width / pooled_height) % channels; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
120 int n = index / pooled_width / pooled_height / channels; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
142 const int width, const int channels, const int pooled_height, in MaxPoolForwardNHWC() argument
152 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolForwardNHWC()
[all …]
Dpooling_ops_3d_gpu.cu.cc32 const int pooled_plane, const int pooled_height, const int pooled_width, in MaxPoolGradBackwardNoMaskNCDHW() argument
41 int ph = (index / pooled_width) % pooled_height; in MaxPoolGradBackwardNoMaskNCDHW()
42 int pp = (index / pooled_width / pooled_height) % pooled_plane; in MaxPoolGradBackwardNoMaskNCDHW()
43 int c = (index / pooled_width / pooled_height / pooled_plane) % channels; in MaxPoolGradBackwardNoMaskNCDHW()
44 int n = (index / pooled_width / pooled_height / pooled_plane / channels); in MaxPoolGradBackwardNoMaskNCDHW()
82 const int pooled_plane, const int pooled_height, const int pooled_width, in MaxPoolGradBackwardNoMaskNDHWC() argument
97 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNDHWC()
100 n /= pooled_height; in MaxPoolGradBackwardNoMaskNDHWC()
137 const int batch, const int pooled_plane, const int pooled_height, in operator ()() argument
144 batch * channels * pooled_plane * pooled_height * pooled_width; in operator ()()
[all …]
Davgpooling_op_gpu.cu.cc46 const int channels, const int pooled_height, in DEFINE_GPU_KERNELS()
59 const int phend = min(h / stride_h + 1, pooled_height); in DEFINE_GPU_KERNELS()
64 top_diff + n * pooled_height * pooled_width * channels + c; in DEFINE_GPU_KERNELS()
86 const int channels, const int pooled_height, in RunAvePoolBackwardNHWC() argument
97 channels, pooled_height, pooled_width, kernel_h, kernel_w, stride_h, in RunAvePoolBackwardNHWC()
105 const int width, const int channels, const int pooled_height,
111 const int width, const int channels, const int pooled_height,
117 const int width, const int channels, const int pooled_height,
Dmaxpooling_op_gpu.h38 const int width, const int channels, const int pooled_height,
48 const int width, int channels, const int pooled_height,
66 const int width, const int channels, const int pooled_height,
85 const int pooled_height, const int pooled_width,
Dpooling_ops_3d_gpu.h36 const int pooled_height, const int pooled_width,
Davgpooling_op.h70 const int channels, const int pooled_height,