Searched refs:window_planes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 68 window_planes = GetTensorDim(ksize, data_format, '0'); in Pool3dParameters()
84 OP_REQUIRES_OK(context, GetWindowedOutputSize(tensor_in_planes, window_planes, in Pool3dParameters()
585 const int32 window_planes = params.window_planes; in launch() local
612 const int p_end = std::min(p_start + window_planes, in_planes); in launch()
649 params.window_planes * params.window_rows * params.window_cols; in launch()
813 params.window_planes, params.window_rows, params.window_cols, in launch()
|
| D | pooling_ops_3d.h | 57 int window_planes; member
|
| D | mkl_pooling_ops_common.cc | 322 window_planes = GetTensorDim(ksize, data_format, '0'); in Init()
337 (window_rows == 1 && window_cols == 1 && window_planes == 1)), in Init()
346 context, GetWindowedOutputSizeVerbose(tensor_in_planes, window_planes, in Init()
|
| D | mkl_pooling_ops_common.h | 372 int window_planes; // Pool3D member
402 window_planes(0), in MklPoolParameters()
538 memory::dims({pool_params->window_planes, pool_params->window_rows, in PoolParamsToDims()
|
| D | pooling_ops_3d_sycl.h | 73 window_planes_(params.window_planes), in SYCL3DPoolParams()
|