torch-mlir/lib/Conversion/Utils
JianzheXiao a42d4c18ff
[Torch Dialect]Support aten.cosine_similarity (#2364)
As title, add support for aten.cosine_similarity, support broadcast
inputA/inputB to the same shape
2023-11-08 15:28:30 +08:00
..
CMakeLists.txt build: update llvm tag to 6f46ff37 (#1448) 2022-10-05 08:28:06 -05:00
Utils.cpp [Torch Dialect]Support aten.cosine_similarity (#2364) 2023-11-08 15:28:30 +08:00