Searched refs:TFDataFormatIsNHWC (Results 1 – 3 of 3) sorted by relevance
33 inline bool TFDataFormatIsNHWC(Operation *op) { in TFDataFormatIsNHWC() function
95 if (!TFTypeIsFloat32Tensor(op.input()) || !TFDataFormatIsNHWC(op)) in matchAndRewrite()
321 if (!TFDataFormatIsNHWC(op)) return failure(); in matchAndRewrite()1038 if (!TFDataFormatIsNHWC(fused_batch_norm_op) && in matchAndRewrite()