| /external/tensorflow/tensorflow/core/kernels/ |
| pooling_ops_3d.h | 60 int depth_window; member in struct:tensorflow::Pool3dParameters
|
| avgpooling_op.cc | 83 OP_REQUIRES(context, params.depth_window == 1,
148 OP_REQUIRES(context, params.depth_window == 1,
273 const int depth_window = ksize_[3]; variable
281 // Spatial pooling is when depth_window = 1
282 OP_REQUIRES(context, depth_window == 1,
348 window_rows * window_cols * depth_window * in_rows * in_rows * in_cols;
513 const int depth_window = ksize_[3]; variable
521 // Spatial pooling is when depth_window = 1
522 OP_REQUIRES(context, depth_window == 1,
|
| pooling_ops_common.h | 58 int depth_window; member in struct:tensorflow::PoolParameters
121 if (params.depth_window > 1) {
124 OP_REQUIRES(context, params.depth % params.depth_window == 0,
129 context, params.depth_window == params.depth_stride,
149 in_by_pool(tensor_in.flat<T>().data(), params.depth_window,
150 tensor_in.NumElements() / params.depth_window);
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...] |
| mkl_pooling_ops_common.h | 375 int depth_window; member in struct:tensorflow::MklPoolParameters
405 depth_window(0),
|
