Searched refs:Dim3d_H (Results 1 – 4 of 4) sorted by relevance
168 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_rows; in GetInputSizeInMklOrder()269 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = filter_rows; in GetFilterSizeInMklOrder()466 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = static_cast<int>(out_rows);
523 bwd_output_dims[MklDnnDims3D::Dim3d_H], in Compute()
631 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_md.data.dims[3]; in ConfigureInput()
108 Dim3d_H = 3, enumerator551 return MklDnnDims3D::Dim3d_H; in GetMklDnnTensor3DDimIndex()683 MklDnnDims3D::Dim3d_H; in SetTfDimOrder()