diff --git a/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp b/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp index a2d58daac..ff2728764 100644 --- a/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp +++ b/lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp @@ -1273,13 +1273,13 @@ public: // Set the values in the input tensor to the smallest element of that // type TypedAttr minAttr = getNumericLimit(rewriter, srcType.getElementType(), - /*getMin=*/true); + /*getMin=*/true); normalizationValue = rewriter.create(loc, minAttr); } else if (reduceEnum == torch_upstream::ReductionType::MIN) { // Set the values in the input tensor to the largest element of that // type TypedAttr maxAttr = getNumericLimit(rewriter, srcType.getElementType(), - /*getMin=*/false); + /*getMin=*/false); normalizationValue = rewriter.create(loc, maxAttr); } @@ -1551,10 +1551,10 @@ public: Value result = createTMTensorScanOp( rewriter, loc, input, output, acc, dim, /*inclusive=*/true, [](OpBuilder &b, Location loc, Value input, Value acc) { - Value sum = (input.getType().isa() - ? b.create(loc, input, acc) - : b.create(loc, input, acc)) - ->getResult(0); + Value sum = + (input.getType().isa() + ? b.create(loc, input, acc)->getResult(0) + : b.create(loc, input, acc)->getResult(0)); b.create(loc, sum); });