Searched refs:depth_window (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.h | 58 int depth_window; member
121 if (params.depth_window > 1) { in Compute()
124 OP_REQUIRES(context, params.depth % params.depth_window == 0, in Compute()
129 context, params.depth_window == params.depth_stride, in Compute()
149 in_by_pool(tensor_in.flat<T>().data(), params.depth_window, in DepthwiseMaxPool()
150 tensor_in.NumElements() / params.depth_window); in DepthwiseMaxPool()
360 if (params.depth_window > 1) {
363 OP_REQUIRES(context, params.depth % params.depth_window == 0,
368 context, params.depth_window == params.depth_stride,
388 in_by_pool(tensor_in.flat<T>().data(), params.depth_window,
[all …]
|
| D | pooling_ops_common.cc | 68 depth_window = GetTensorDim(ksize, data_format, 'C'); in PoolParameters()
76 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in PoolParameters()
81 if (depth_window == 1) { in PoolParameters()
95 context, depth % depth_window == 0, in PoolParameters()
99 context, depth_stride == depth_window, in PoolParameters()
112 out_depth = depth / depth_window; in PoolParameters()
117 if (depth_window == 1) { in forward_output_shape()
|
| D | mkl_pooling_ops_common.cc | 305 depth_window = GetTensorDim(ksize, data_format, 'C'); in Init()
315 (depth_window == 1 || (window_rows == 1 && window_cols == 1)), in Init()
325 depth_window = GetTensorDim(ksize, data_format, 'C'); in Init()
336 (depth_window == 1 || in Init()
343 if (depth_window == 1) { // we are pooling in the D (Pool3D only), H and W in Init()
382 OP_REQUIRES(context, depth % depth_window == 0, in Init()
386 OP_REQUIRES(context, depth_stride == depth_window, in Init()
399 out_depth = depth / depth_window; in Init()
|
| D | avgpooling_op.cc | 83 OP_REQUIRES(context, params.depth_window == 1, in Compute()
148 OP_REQUIRES(context, params.depth_window == 1, in Compute()
273 const int depth_window = ksize_[3]; in Compute() local
282 OP_REQUIRES(context, depth_window == 1, in Compute()
348 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols; in Compute()
513 const int depth_window = ksize_[3]; in Compute() local
522 OP_REQUIRES(context, depth_window == 1, in Compute()
|
| D | pooling_ops_3d.h | 60 int depth_window; member
|
| D | quantized_pooling_ops.cc | 66 OP_REQUIRES(context, params.depth_window == 1, in Compute()
|
| D | mkl_pooling_ops_common.h | 375 int depth_window; member
405 depth_window(0), in MklPoolParameters()
|
| D | mkl_maxpooling_op.cc | 89 context, (pool_params.depth_window == 1), in Compute()
256 context, (pool_params.depth_window == 1), in Compute()
|
| D | pooling_ops_3d.cc | 71 depth_window = GetTensorDim(ksize, data_format, 'C'); in Pool3dParameters()
80 context, depth_window == 1 && depth_stride == 1, in Pool3dParameters()
|