Searched refs:IsMklChannelDim (Results 1 – 2 of 2) sorted by relevance
111 } else if (!src_dnn_shape.IsMklChannelDim(src_dnn_shape.GetDimension() - in Compute()357 input_grad_dnn_shape.IsMklChannelDim( in Compute()359 orig_input_dnn_shape.IsMklChannelDim( in Compute()361 orig_output_dnn_shape.IsMklChannelDim( in Compute()
532 inline bool IsMklChannelDim(int d) const { in IsMklChannelDim() function558 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNCHWFormat()568 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNHWCFormat()