Searched refs:Dim_N (Results 1 – 3 of 3) sorted by relevance
147 mkldnn_sizes[MklDnnDims::Dim_N] = input_batch; in GetInputSizeInMklOrder()465 mkldnn_sizes[MklDnnDims::Dim_N] = out_batch;
642 (src_dims[MklDnnDims::Dim_N] > kSmallBatchSize) && in Compute()
71 Dim_N = 0, enumerator338 return MklDnnDims::Dim_N; in GetMklDnnTensorDimIndex()453 data_.map_[0] = MklDnnDims::Dim_N; in SetTfLayout()515 data_.map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDnnDims::Dim_N; in SetTfDimOrder()539 return TfDimIdx(d) == MklDnnDims::Dim_N; in IsMklBatchDim()