Searched refs:IsMklBatchDim (Results 1 – 1 of 1) sorted by relevance
538 inline bool IsMklBatchDim(int d) const { in IsMklBatchDim() function557 return (IsMklBatchDim(GetTensorDimIndex<2>(data_format, 'N')) && in IsTensorInNCHWFormat()567 return (IsMklBatchDim(GetTensorDimIndex<2>(data_format, 'N')) && in IsTensorInNHWCFormat()