| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op_gpu.cu.cc | 47 const int pooled_width, const int kernel_h, in DEFINE_GPU_KERNELS()
58 const int phstart = (h < kernel_h) ? 0 : (h - kernel_h) / stride_h + 1; in DEFINE_GPU_KERNELS()
70 int hend = min(hstart + kernel_h, height); 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,
|
| D | maxpooling_op_gpu.cu.cc | 69 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNCHW() argument
79 int hend = min(hstart + kernel_h, height); in MaxPoolForwardNCHW()
111 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
123 int hend = min(hstart + kernel_h, height); in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
143 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolForwardNHWC() argument
154 int hend = min(hstart + kernel_h, height); in MaxPoolForwardNHWC()
182 const int pooled_width, const int kernel_h, const int kernel_w, in MaxPoolBackwardNoMaskNHWC() argument
194 int hend = min(hstart + kernel_h, height); in MaxPoolBackwardNoMaskNHWC()
269 const int height, const int width, const int kernel_h, const int kernel_w, in MaxPoolGradBackwardNoMaskNCHW() argument
280 const int hend = min(hstart + kernel_h, height); 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
49 const int hend = min(hstart + kernel_h, height); in MaxPoolGradBackwardNoMaskNCDHW()
84 const int kernel_p, const int kernel_h, const int kernel_w, in MaxPoolGradBackwardNoMaskNDHWC() argument
98 int hend = min(hstart + kernel_h, height); in MaxPoolGradBackwardNoMaskNDHWC()
139 const int height, const int width, const int kernel_p, const int kernel_h, in operator ()() argument
150 pooled_width, channels, plane, height, width, kernel_p, kernel_h, in operator ()()
157 pooled_width, channels, plane, height, width, kernel_p, kernel_h, in operator ()()
|
| D | maxpooling_op_gpu.h | 39 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,
87 const int kernel_h, const int kernel_w, const int stride_h,
|
| D | pooling_ops_3d_gpu.h | 38 const int width, const int kernel_p, const int kernel_h,
|
| D | avgpooling_op.h | 71 const int pooled_width, const int kernel_h,
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | convolutional.py | 787 kernel_h, kernel_w = self.kernel_size
797 kernel_h,
846 kernel_h, kernel_w = self.kernel_size
857 kernel_h,
1063 kernel_d, kernel_h, kernel_w = self.kernel_size
1078 kernel_h,
1128 kernel_d, kernel_h, kernel_w = self.kernel_size
1145 kernel_h,
|
| /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]
|