Searched refs:IsMklWidthDim (Results 1 – 1 of 1) sorted by relevance
259 bool IsMklWidthDim(int d) const { return tf_dim_idx(d) == MklDims::W; } in IsMklWidthDim() function271 IsMklWidthDim(GetTensorDimIndex<2>(data_format, 'W'))); in IsTensorInNCHWFormat()281 IsMklWidthDim(GetTensorDimIndex<2>(data_format, 'W'))); in IsTensorInNHWCFormat()723 inline bool IsMklWidthDim(int d) const { in IsMklWidthDim() function739 IsMklWidthDim(GetTensorDimIndex<2>(data_format, 'W'))); in IsTensorInNCHWFormat()749 IsMklWidthDim(GetTensorDimIndex<2>(data_format, 'W'))); in IsTensorInNHWCFormat()