Searched refs:Dim3d_N (Results 1 – 3 of 3) sorted by relevance
171 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_batch; in GetInputSizeInMklOrder()472 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = out_batch;
630 mkldnn_sizes[MklDnnDims3D::Dim3d_N] = input_md.data.dims[0]; in ConfigureInput()
83 Dim3d_N = 0, enumerator354 return MklDnnDims3D::Dim3d_N; in GetMklDnnTensor3DDimIndex()508 MklDnnDims3D::Dim3d_N; in SetTfDimOrder()