Searched refs:Dim3d_N (Results 1 – 3 of 3) sorted by relevance
165 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_batch; in GetInputSizeInMklOrder()463 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = out_batch;
628 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_md.data.dims[0]; in ConfigureInput()
105 Dim3d_N = 0, enumerator545 return MklDnnDims3D::Dim3d_N; in GetMklDnnTensor3DDimIndex()689 MklDnnDims3D::Dim3d_N; in SetTfDimOrder()