torch-mlir

Commit Graph

Author	SHA1	Message	Date
Thomas Dietert	d2432bbe5a	[MLIR][Torch] Do not convert bias tensor to element type if NoneType (#3072 ) The `convertTensorToElementType` function expects it's argument to have a valid tensor type that is not `Torch::NoneType`. This PR checks that the bias tensor is not of type `Torch::NoneType` before calling `convertTensorToElementType` on the bias tensor argument in the `matchAndRewrite` member function of the `ConvertAtenConvolutionOp` class.	2024-04-02 14:19:26 +05:30

Author

SHA1

Message

Date

Thomas Dietert

d2432bbe5a

[MLIR][Torch] Do not convert bias tensor to element type if NoneType (#3072 )

The `convertTensorToElementType` function expects it's argument to have
a valid tensor type that is not `Torch::NoneType`. This PR checks that
the bias tensor is not of type `Torch::NoneType` before calling
`convertTensorToElementType` on the bias tensor argument in the
`matchAndRewrite` member function of the `ConvertAtenConvolutionOp`
class.

2024-04-02 14:19:26 +05:30

1 Commits (ce7d4f1660c210b98978b73a0ed4a3db440e4646)