torch-mlir/lib/Conversion/TorchOnnxToTorch
aldesilv a4ba02eef5
[ONNX] add support for tfidfvectorizer (#3553)
1-d/2-d input and output
implemented based on the description and example test cases in
https://github.com/onnx/onnx/blob/main/docs/Operators.md#TfIdfVectorizer
and some notes from

https://github.com/onnx/onnx/blob/main/onnx/reference/ops/op_tfidf_vectorizer.py#L128

---------

Co-authored-by: zjgarvey <zjgarvey@gmail.com>
2024-08-12 18:10:11 -05:00
..
CMakeLists.txt Support onnx.GRU and onnx.RNN (#3447) 2024-07-10 14:04:17 -04:00
DefaultDomainAtoF.cpp [onnx] Add support for `fp8` `onnx.DequantizeLinear` (#3617) 2024-08-08 16:20:53 -07:00
DefaultDomainGtoP.cpp [onnx] Fix `onnx.Hardmax` lowering to torch (#3624) 2024-08-12 11:19:02 -07:00
DefaultDomainQtoZ.cpp [ONNX] add support for tfidfvectorizer (#3553) 2024-08-12 18:10:11 -05:00
OnnxRecurrentLayerOpExpanders.cpp Support onnx.GRU and onnx.RNN (#3447) 2024-07-10 14:04:17 -04:00
PassDetail.h Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Passes.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Patterns.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
TorchOnnxToTorch.cpp Initial TorchOnnxToTorch conversion pipeline. (#2585) 2023-11-21 21:02:55 -08:00
Utils.cpp Support onnx.GRU and onnx.RNN (#3447) 2024-07-10 14:04:17 -04:00