Searched refs:tensor_in_planes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.h | 52 int tensor_in_planes; member
|
| D | pooling_ops_3d.cc | 59 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Pool3dParameters()
79 OP_REQUIRES_OK(context, GetWindowedOutputSize(tensor_in_planes, window_planes, in Pool3dParameters()
577 params.tensor_in_planes * params.tensor_in_cols * in launch()
585 params.tensor_in_planes * params.tensor_in_cols * in launch()
598 const int32_t in_planes = params.tensor_in_planes; in launch()
864 params.tensor_in_planes, params.tensor_in_rows, params.tensor_in_cols, in launch()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.cc | 246 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Init()
273 tensor_in_planes = mklInputShape->GetDimension3D('D'); in Init()
335 context, GetWindowedOutputSizeVerbose(tensor_in_planes, window_planes, in Init()
|
| D | mkl_pooling_ops_common.h | 369 int tensor_in_planes; // Pool3D member
400 tensor_in_planes(0), in MklPoolParameters()
|