torch-mlir/frontends/pytorch/csrc/builder
Bairen Yi fead0312f1 Revert "Also fallback autograd dispatch keys for torchvision::nms"
This reverts commit 30a42dea32.
2021-03-16 19:37:45 -07:00
..
acap_dispatch.cpp Revert "Also fallback autograd dispatch keys for torchvision::nms" 2021-03-16 19:37:45 -07:00
acap_dispatch.h Make torch_mlir compatible with binary PyTorch installations. 2020-12-14 09:51:00 -08:00
class_annotator.cpp Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
class_annotator.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
debug.cpp Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
debug.h Delete old PyTorch 1.3 type dispatch oriented code paths. 2020-11-12 22:27:05 -08:00
func_builder.cpp Add support for prim::GetAttr/SetAttr/CallMethod/If 2021-02-04 17:01:47 -08:00
func_builder.h Add initial TorchScript module importer 2021-01-28 11:55:17 -08:00
function_importer.cpp Properly model "derefinement". 2021-03-03 15:09:44 -08:00
function_importer.h Properly model "derefinement". 2021-03-03 15:09:44 -08:00
ivalue_importer.cpp [torch] Add import support for IValue string Type(s) (#179) 2021-03-04 13:08:50 -08:00
ivalue_importer.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
mlir_utils.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
module_builder.cpp Bump llvm-project to e31c77b1827fa4dd3511f21af11cfab18ecf6d38 2021-03-10 11:01:16 -08:00
module_builder.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
node_importer.cpp Add prim::device and handle derefining for prim::CallMethod 2021-03-11 14:10:09 -08:00
node_importer.h Properly model "derefinement". 2021-03-03 15:09:44 -08:00
op_builder.cpp Add prim::Print and fix prim::CallMethod 2021-02-10 15:15:56 -08:00
op_builder.h Add prim::Print and fix prim::CallMethod 2021-02-10 15:15:56 -08:00
python_bindings.cpp Make ATenOpRegistrations.txt dump more readable. 2021-03-16 12:39:28 -07:00
torch_to_mlir_utils.cpp Add prim::device and handle derefining for prim::CallMethod 2021-03-11 14:10:09 -08:00
torch_to_mlir_utils.h Properly model "derefinement". 2021-03-03 15:09:44 -08:00