Searched refs:Dim_H (Results 1 – 3 of 3) sorted by relevance
143 mkldnn_sizes[MklDnnDims::Dim_H] = input_rows; in GetInputSizeInMklOrder()240 mkldnn_sizes[MklDnnDims::Dim_H] = filter_rows; in GetFilterSizeInMklOrder()458 mkldnn_sizes[MklDnnDims::Dim_H] = static_cast<int>(out_rows);
491 {bwd_output_dims[MklDnnDims::Dim_H], in Compute()681 TensorShape output_tf_shape({output_dims_mkl_order[MklDnnDims::Dim_H], in AllocateOutputTensor()
95 Dim_H = 2, enumerator533 return MklDnnDims::Dim_H; in GetMklDnnTensorDimIndex()694 data_.map_[GetTensorDimIndex<2>(data_format, 'H')] = MklDnnDims::Dim_H; in SetTfDimOrder()729 return TfDimIdx(d) == MklDnnDims::Dim_H; in IsMklHeightDim()