Home
last modified time | relevance | path

Searched refs:Dim3d_C (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h172 mkldnn_sizes[MklDnnDims3D::Dim3d_C] = input_depth; in GetInputSizeInMklOrder()
473 mkldnn_sizes[MklDnnDims3D::Dim3d_C] = out_depth;
Dmkl_pooling_ops_common.h631 mkldnn_sizes[MklDnnDims3D::Dim3d_C] = input_md.data.dims[1]; in ConfigureInput()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h84 Dim3d_C = 1, enumerator
356 return MklDnnDims3D::Dim3d_C; in GetMklDnnTensor3DDimIndex()
506 MklDnnDims3D::Dim3d_C; in SetTfDimOrder()