Searched refs:IsMklWidthDim (Results 1 – 1 of 1) sorted by relevance
544 inline bool IsMklWidthDim(int d) const { in IsMklWidthDim() function560 IsMklWidthDim(GetTensorDimIndex<2>(data_format, 'W'))); in IsTensorInNCHWFormat()570 IsMklWidthDim(GetTensorDimIndex<2>(data_format, 'W'))); in IsTensorInNHWCFormat()