mirror of https://github.com/llvm/torch-mlir
d2432bbe5a
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. |
||
|---|---|---|
| .. | ||
| basic.mlir | ||
| broadcast.mlir | ||
| convolution.mlir | ||
| elementwise.mlir | ||
| flatten.mlir | ||
| gridsampler.mlir | ||
| pooling.mlir | ||
| sparse.mlir | ||
| unsqueeze.mlir | ||
| view.mlir | ||