Searched defs:IsMklBatchDim (Results 1 – 1 of 1) sorted by relevance
256 bool IsMklBatchDim(int d) const { return tf_dim_idx(d) == MklDims::N; } in IsMklBatchDim() function718 inline bool IsMklBatchDim(int d) const { in IsMklBatchDim() function