Searched refs:Dim_H (Results 1 – 3 of 3) sorted by relevance
149 mkldnn_sizes[MklDnnDims::Dim_H] = input_rows; in GetInputSizeInMklOrder()246 mkldnn_sizes[MklDnnDims::Dim_H] = filter_rows; in GetFilterSizeInMklOrder()467 mkldnn_sizes[MklDnnDims::Dim_H] = static_cast<int>(out_rows);
489 {diff_filter_dims[MklDnnDims::Dim_H], in Compute()689 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor()
73 Dim_H = 2, enumerator342 return MklDnnDims::Dim_H; in GetMklDnnTensorDimIndex()513 data_.map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDnnDims::Dim_H; in SetTfDimOrder()550 return TfDimIdx(d) == MklDnnDims::Dim_H; in IsMklHeightDim()