Searched refs:IsMklChannelDim (Results 1 – 2 of 2) sorted by relevance
109 } else if (!src_dnn_shape.IsMklChannelDim(src_dnn_shape.GetDimension() - in Compute()361 input_grad_dnn_shape.IsMklChannelDim( in Compute()363 orig_input_dnn_shape.IsMklChannelDim( in Compute()365 orig_output_dnn_shape.IsMklChannelDim( in Compute()
253 bool IsMklChannelDim(int d) const { return tf_dim_idx(d) == MklDims::C; } in IsMklChannelDim() function269 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNCHWFormat()279 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNHWCFormat()713 inline bool IsMklChannelDim(int d) const { in IsMklChannelDim() function737 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNCHWFormat()747 IsMklChannelDim(GetTensorDimIndex<2>(data_format, 'C')) && in IsTensorInNHWCFormat()