Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmaxpooling_op_gpu.cu.cc69 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNCHW() argument
73 int pw = index % pooled_width; in MaxPoolForwardNCHW()
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()
111 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
117 int pw = index % pooled_width; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
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()
[all …]
Dpooling_ops_3d_gpu.cu.cc32 const int pooled_plane, const int pooled_height, const int pooled_width, in MaxPoolGradBackwardNoMaskNCDHW() argument
40 int pw = index % pooled_width; in MaxPoolGradBackwardNoMaskNCDHW()
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
93 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNDHWC()
96 n /= pooled_width; in MaxPoolGradBackwardNoMaskNDHWC()
138 const int pooled_width, const int channels, const int plane, in operator ()() argument
[all …]
Davgpooling_op_gpu.cu.cc47 const int pooled_width, const int kernel_h, in DEFINE_GPU_KERNELS()
61 const int pwend = min(w / stride_w + 1, pooled_width); in DEFINE_GPU_KERNELS()
64 top_diff + n * pooled_height * pooled_width * channels + c; in DEFINE_GPU_KERNELS()
75 gradient += top_diff_slice[(ph * pooled_width + pw) * channels] / in DEFINE_GPU_KERNELS()
87 const int pooled_width, const int kernel_h, in RunAvePoolBackwardNHWC() argument
97 channels, pooled_height, pooled_width, kernel_h, kernel_w, stride_h, in RunAvePoolBackwardNHWC()
106 const int pooled_width, const int kernel_h, const int kernel_w,
112 const int pooled_width, const int kernel_h, const int kernel_w,
118 const int pooled_width, const int kernel_h, const int kernel_w,
Dmaxpooling_op_gpu.h39 const int pooled_width, const int kernel_h,
49 const int pooled_width, const int kernel_h,
67 const int pooled_width, const int kernel_h,
85 const int pooled_height, const int pooled_width,
Dpooling_ops_3d_gpu.h36 const int pooled_height, const int pooled_width,
Davgpooling_op.h71 const int pooled_width, const int kernel_h,