| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op_gpu.cu.cc | 49 const int stride_w, const int pad_t, in DEFINE_GPU_KERNELS()
56 const int h = (index / channels / width) % height + pad_t; in DEFINE_GPU_KERNELS()
68 int hstart = ph * stride_h - pad_t; in DEFINE_GPU_KERNELS()
89 const int stride_w, const int pad_t, in RunAvePoolBackwardNHWC() argument
98 stride_w, pad_t, pad_t, bottom_diff)); in RunAvePoolBackwardNHWC()
107 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
113 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
119 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
|
| 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() argument
77 int hstart = ph * stride_h - pad_t; in MaxPoolForwardNCHW()
112 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
121 int hstart = ph * stride_h - pad_t; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
144 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC() argument
152 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolForwardNHWC()
183 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolBackwardNoMaskNHWC() argument
192 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolBackwardNoMaskNHWC()
270 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW() argument
278 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCHW()
[all …]
|
| D | pooling_ops_3d_gpu.cu.cc | 36 const int pad_t, const int pad_l, const dtype* top_diff, in MaxPoolGradBackwardNoMaskNCDHW() argument
46 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCDHW()
86 const int pad_t, const int pad_l, const dtype* top_diff, in MaxPoolGradBackwardNoMaskNDHWC() argument
97 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNDHWC()
141 const int stride_w, const int pad_p, const int pad_t, const int pad_l, in operator ()() argument
151 kernel_w, stride_p, stride_h, stride_w, pad_p, pad_t, pad_l, top_diff, in operator ()()
158 kernel_w, stride_p, stride_h, stride_w, pad_p, pad_t, pad_l, top_diff, in operator ()()
|
| D | maxpooling_op_gpu.h | 41 const int pad_t, const int pad_l, T* top_data, int64* mask,
51 const int pad_t, const int pad_l, int32* top_data,
69 const int pad_t, const int pad_l, const T* top_diff,
88 const int stride_w, const int pad_t, const int pad_l,
|
| D | pooling_ops_3d_gpu.h | 40 const int stride_w, const int pad_p, const int pad_t,
|
| D | avgpooling_op.h | 73 const int stride_w, const int pad_t,
|
| D | conv_grad_ops_3d.cc | 68 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, in Col2im() argument
72 const int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Col2im()
76 int h_pad = -pad_t; in Col2im()
117 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, in Im2col() argument
121 const int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Im2col()
126 int h_pad = -pad_t; in Im2col()
|
| D | conv_grad_filter_ops.cc | 67 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Im2col() argument
69 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Im2col()
72 int h_pad = -pad_t; in Im2col()
|
| D | conv_grad_input_ops.cc | 67 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im() argument
69 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Col2im()
71 int h_pad = -pad_t; in Col2im()
|