Searched refs:IsMklHeightDim (Results 1 – 1 of 1) sorted by relevance
262 bool IsMklHeightDim(int d) const { return tf_dim_idx(d) == MklDims::H; } in IsMklHeightDim() function270 IsMklHeightDim(GetTensorDimIndex<2>(data_format, 'H')) && in IsTensorInNCHWFormat()280 IsMklHeightDim(GetTensorDimIndex<2>(data_format, 'H')) && in IsTensorInNHWCFormat()728 inline bool IsMklHeightDim(int d) const { in IsMklHeightDim() function738 IsMklHeightDim(GetTensorDimIndex<2>(data_format, 'H')) && in IsTensorInNCHWFormat()748 IsMklHeightDim(GetTensorDimIndex<2>(data_format, 'H')) && in IsTensorInNHWCFormat()