// precise than the input types, are used for this op.
    Type kernel_type = mlir::cast<TensorType>(scale.getType()).getElementType();
    grad = rewriter.create<ConvertOp>(loc, grad, kernel_type);
    act = rewriter.create<ConvertOp>(loc, act, kernel_type);

    tensorflow::TensorFormat data_format;
    if (!FormatFromString(op.getDataFormat().str(), &data_format))