Searched refs:Dim3d_H (Results 1 – 4 of 4) sorted by relevance
174 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_rows; in GetInputSizeInMklOrder()275 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = filter_rows; in GetFilterSizeInMklOrder()475 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = static_cast<int>(out_rows);
522 diff_filter_dims[MklDnnDims3D::Dim3d_H], in Compute()
633 mkldnn_sizes[MklDnnDims3D::Dim3d_H] = input_md.data.dims[3]; in ConfigureInput()
86 Dim3d_H = 3, enumerator360 return MklDnnDims3D::Dim3d_H; in GetMklDnnTensor3DDimIndex()502 MklDnnDims3D::Dim3d_H; in SetTfDimOrder()