Searched refs:Dim_N (Results 1 – 3 of 3) sorted by relevance
141 mkldnn_sizes[MklDnnDims::Dim_N] = input_batch; in GetInputSizeInMklOrder()456 mkldnn_sizes[MklDnnDims::Dim_N] = out_batch;
1061 (src_dims[MklDnnDims::Dim_N] > kSmallBatchSize) && in Compute()
93 Dim_N = 0, enumerator529 return MklDnnDims::Dim_N; in GetMklDnnTensorDimIndex()696 data_.map_[GetTensorDimIndex<2>(data_format, 'N')] = MklDnnDims::Dim_N; in SetTfDimOrder()719 return TfDimIdx(d) == MklDnnDims::Dim_N; in IsMklBatchDim()