File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
tensorflow/compiler/mlir/quantization/stablehlo/passes Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -219,7 +219,7 @@ class InsertWeightParamPattern
219
219
dimension_numbers.getRhsContractingDimensions ();
220
220
ArrayRef<int64_t > rhs_batching_dims =
221
221
dimension_numbers.getRhsBatchingDimensions ();
222
- int64_t rank = dot.getRhs ().getType (). cast <TensorType>( ).getRank ();
222
+ int64_t rank = cast<TensorType>( dot.getRhs ().getType ()).getRank ();
223
223
for (int i = 0 ; i < rank; ++i) {
224
224
// Return the first non-contracting, non-batching dimension of rhs.
225
225
if (llvm::find (rhs_contracting_dims, i) == rhs_contracting_dims.end () &&
@@ -228,7 +228,7 @@ class InsertWeightParamPattern
228
228
}
229
229
}
230
230
}
231
- return op.getOperand (1 ).getType (). cast <TensorType>( ).getRank () - 1 ;
231
+ return cast<TensorType>( op.getOperand (1 ).getType ()).getRank () - 1 ;
232
232
}
233
233
};
234
234
You can’t perform that action at this time.
0 commit comments