Searched refs:Dim3d_D (Results 1 – 4 of 4) sorted by relevance
173 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_planes; in GetInputSizeInMklOrder()274 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = filter_planes; in GetFilterSizeInMklOrder()474 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = static_cast<int>(out_planes);
521 {diff_filter_dims[MklDnnDims3D::Dim3d_D], in Compute()
632 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_md.data.dims[2]; in ConfigureInput()
85 Dim3d_D = 2, enumerator358 return MklDnnDims3D::Dim3d_D; in GetMklDnnTensor3DDimIndex()500 MklDnnDims3D::Dim3d_D; in SetTfDimOrder()