| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op_gpu.cu.cc | 47 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in DEFINE_GPU_KERNELS()
54 const int h = (index / channels / width) % height + pad_t; in DEFINE_GPU_KERNELS()
66 int hstart = ph * stride_h - pad_t; in DEFINE_GPU_KERNELS()
87 const int stride_w, const int pad_t, in RunAvePoolBackwardNHWC() argument
96 stride_w, pad_t, pad_t, bottom_diff)); in RunAvePoolBackwardNHWC()
105 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
111 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
117 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
|
| D | maxpooling_op_gpu.cu.cc | 71 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in MaxPoolForwardNCHW() argument
79 int hstart = ph * stride_h - pad_t; in MaxPoolForwardNCHW()
115 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
124 int hstart = ph * stride_h - pad_t; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
148 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC() argument
157 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolForwardNHWC()
235 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW() argument
243 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCHW()
276 const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNHWC() argument
285 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNHWC()
[all …]
|
| D | pooling_ops_3d_gpu.cu.cc | 36 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNCDHW() argument
47 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCDHW()
87 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNDHWC() argument
99 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNDHWC()
143 const int stride_w, const int pad_p, const int pad_t, const int pad_l, in operator ()() argument
154 stride_w, pad_p, pad_t, pad_l, top_diff, bottom_diff)); in operator ()()
161 stride_w, pad_p, pad_t, pad_l, top_diff, bottom_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,
78 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_input_ops.h | 79 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im() argument
81 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Col2im()
83 int h_pad = -pad_t; in Col2im()
|
| D | conv_grad_ops_3d.cc | 77 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, in Col2im() argument
81 const int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Col2im()
85 int h_pad = -pad_t; in Col2im()
126 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, in Im2col() argument
130 const int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Im2col()
135 int h_pad = -pad_t; in Im2col()
|
| D | conv_grad_filter_ops.cc | 72 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Im2col() argument
74 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Im2col()
77 int h_pad = -pad_t; in Im2col()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5664 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im() argument
5667 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Col2im()
5669 int h_pad = -pad_t; in Col2im()
|