| /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() argument
78 int wstart = pw * stride_w - 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() argument
122 int wstart = pw * stride_w - 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() argument
150 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolForwardNHWC()
183 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolBackwardNoMaskNHWC() argument
190 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolBackwardNoMaskNHWC()
270 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW() argument
279 int wstart = pw * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNCHW()
[all …]
|
| D | avgpooling_op_gpu.cu.cc | 50 const int pad_l, dtype* const bottom_diff) { in DEFINE_GPU_KERNELS()
55 const int w = index / channels % width + pad_l; in DEFINE_GPU_KERNELS()
69 int wstart = pw * stride_w - pad_l; in DEFINE_GPU_KERNELS()
90 const int pad_l, T* const bottom_diff, in RunAvePoolBackwardNHWC() argument
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 | pooling_ops_3d_gpu.cu.cc | 36 const int pad_t, const int pad_l, const dtype* top_diff, in MaxPoolGradBackwardNoMaskNCDHW() argument
47 int wstart = pw * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNCDHW()
86 const int pad_t, const int pad_l, const dtype* top_diff, in MaxPoolGradBackwardNoMaskNDHWC() argument
93 int wstart = (n % pooled_width) * stride_w - pad_l; 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 | mkl_conv_ops.h | 316 memory::dims* pad_l, memory::dims* pad_r, bool pad_enabled = false,
320 CHECK_NOTNULL(pad_l);
398 pad_top = static_cast<int64>((*pad_l)[0]);
399 pad_left = static_cast<int64>((*pad_l)[1]);
431 *pad_l = {static_cast<int>(pad_top), static_cast<int>(pad_left)};
436 *pad_l = {static_cast<int>(pad_D1), static_cast<int>(pad_top),
479 memory::dims* output_dims_mkl_order, memory::dims* pad_l, in GetOutputAndPadSizeInMklOrder() argument
483 CHECK_NOTNULL(pad_l); in GetOutputAndPadSizeInMklOrder()
503 output_dims_mkl_order, pad_l, pad_r, in GetOutputAndPadSizeInMklOrder()
520 memory::dims* pad_l, memory::dims* pad_r, bool pad_enabled = false,
[all …]
|
| D | pooling_ops_3d_gpu.h | 41 const int pad_l, const T* top_diff, T* bottom_diff,
|
| D | avgpooling_op.h | 74 const int pad_l, T* const bottom_diff,
|
| 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
73 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
78 int w_pad = -pad_l; in Col2im()
117 const int pad_t, const int pad_l, const int pad_pb, const int pad_b, in Im2col() argument
122 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
128 int w_pad = -pad_l; 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
70 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
74 int w_pad = -pad_l; 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
70 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
73 int w_pad = -pad_l; in Col2im()
|