Searched refs:tensor_in_planes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.h | 52 int tensor_in_planes; member
|
| D | pooling_ops_3d.cc | 64 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Pool3dParameters()
84 OP_REQUIRES_OK(context, GetWindowedOutputSize(tensor_in_planes, window_planes, in Pool3dParameters()
558 params.tensor_in_planes * params.tensor_in_cols * in launch()
566 params.tensor_in_planes * params.tensor_in_cols * in launch()
579 const int32 in_planes = params.tensor_in_planes; in launch()
812 params.tensor_in_planes, params.tensor_in_rows, params.tensor_in_cols, in launch()
|
| D | mkl_pooling_ops_common.cc | 241 tensor_in_planes = GetTensorDim(tensor_in_shape, data_format, '0'); in Init()
284 tensor_in_planes = mklInputShape->GetDimension3D('D'); in Init()
346 context, GetWindowedOutputSizeVerbose(tensor_in_planes, window_planes, in Init()
|
| D | mkl_pooling_ops_common.h | 367 int tensor_in_planes; // Pool3D member
398 tensor_in_planes(0), in MklPoolParameters()
|
| D | pooling_ops_3d_sycl.h | 70 in_planes_(params.tensor_in_planes), in SYCL3DPoolParams()
|