| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op_gpu.cu.cc | 48 const int kernel_w, const int stride_h, in DEFINE_GPU_KERNELS()
60 const int pwstart = (w < kernel_w) ? 0 : (w - kernel_w) / stride_w + 1; in DEFINE_GPU_KERNELS()
71 int wend = min(wstart + kernel_w, width); in DEFINE_GPU_KERNELS()
88 const int kernel_w, const int stride_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,
|
| D | maxpooling_op_gpu.cu.cc | 69 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNCHW() argument
80 int wend = min(wstart + kernel_w, width); in MaxPoolForwardNCHW()
111 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
124 int wend = min(wstart + kernel_w, width); in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
143 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNHWC() argument
155 int wend = min(wstart + kernel_w, width); in MaxPoolForwardNHWC()
182 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolBackwardNoMaskNHWC() argument
195 int wend = min(wstart + kernel_w, width); in MaxPoolBackwardNoMaskNHWC()
269 const int height, const int width, const int kernel_h, const int kernel_w, in MaxPoolGradBackwardNoMaskNCHW() argument
281 const int wend = min(wstart + kernel_w, width); in MaxPoolGradBackwardNoMaskNCHW()
[all …]
|
| D | pooling_ops_3d_gpu.cu.cc | 34 const int kernel_p, const int kernel_h, const int kernel_w, in MaxPoolGradBackwardNoMaskNCDHW() argument
50 const int wend = min(wstart + kernel_w, width); in MaxPoolGradBackwardNoMaskNCDHW()
84 const int kernel_p, const int kernel_h, const int kernel_w, in MaxPoolGradBackwardNoMaskNDHWC() argument
94 int wend = min(wstart + kernel_w, width); in MaxPoolGradBackwardNoMaskNDHWC()
140 const int kernel_w, const int stride_p, const int stride_h, 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 | 40 const int kernel_w, const int stride_h, const int stride_w,
50 const int kernel_w, const int stride_h, const int stride_w,
68 const int kernel_w, const int stride_h, const int stride_w,
87 const int kernel_h, const int kernel_w, const int stride_h,
|
| D | pooling_ops_3d_gpu.h | 39 const int kernel_w, const int stride_p, const int stride_h,
|
| D | avgpooling_op.h | 72 const int kernel_w, const int stride_h,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional.py | 787 kernel_h, kernel_w = self.kernel_size
803 kernel_w,
846 kernel_h, kernel_w = self.kernel_size
864 kernel_w,
1063 kernel_d, kernel_h, kernel_w = self.kernel_size
1083 kernel_w,
1128 kernel_d, kernel_h, kernel_w = self.kernel_size
1151 kernel_w,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 1267 kernel_h, kernel_w = utils.two_element_tuple(kernel_size)
1270 weights_shape = [kernel_h, kernel_w, 1, 1]
2797 kernel_h, kernel_w = utils.two_element_tuple(kernel_size)
2804 depthwise_shape = [kernel_h, kernel_w, num_filters_in, depth_multiplier]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_test.py | 2704 kernel_w = 3
2707 x = convolutional.conv2d(x, num_outputs, [1, kernel_w])
|