Searched refs:Dim3d_D (Results 1 – 4 of 4) sorted by relevance
167 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_planes; in GetInputSizeInMklOrder()268 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = filter_planes; in GetFilterSizeInMklOrder()465 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = static_cast<int>(out_planes);
522 {bwd_output_dims[MklDnnDims3D::Dim3d_D], in Compute()
630 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_md.data.dims[2]; in ConfigureInput()
107 Dim3d_D = 2, enumerator549 return MklDnnDims3D::Dim3d_D; in GetMklDnnTensor3DDimIndex()681 MklDnnDims3D::Dim3d_D; in SetTfDimOrder()