torch-mlir/frontends/pytorch/csrc/builder
Bryce Arden 1736ff0253 [prim] Add TupleIndex support
I could not find a corresponding ListIndex in prim, which seems to
translate to a __get_attr__ under the hood. I think the reason a tuple
Index op can exist is because Tuple's are supposed to be frozen, where
List operands can be mutable.
2021-03-02 17:28:32 -08:00
..
acap_dispatch.cpp adapt acap_dispatch to latest pytorch nightly ("1.9.0.dev20210215+cpu") 2021-02-18 11:13:29 -08: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
graph_importer.cpp Properly import the entire torch::jit::CompilationUnit 2021-03-01 12:08:01 -08:00
graph_importer.h Add support for prim::GetAttr/SetAttr/CallMethod/If 2021-02-04 17:01:47 -08:00
ivalue_importer.cpp Properly import the entire torch::jit::CompilationUnit 2021-03-01 12:08:01 -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 Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
module_builder.h Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
node_importer.cpp [prim] Add TupleIndex support 2021-03-02 17:28:32 -08:00
node_importer.h Add support for prim::GetAttr/SetAttr/CallMethod/If 2021-02-04 17:01:47 -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 Add ability to annotate TorchScript classes. 2021-02-25 11:28:34 -08:00
torch_to_mlir_utils.cpp Add support for prim::RaiseException. 2021-03-02 16:01:34 -08:00
torch_to_mlir_utils.h Properly import the entire torch::jit::CompilationUnit 2021-03-01 12:08:01 -08:00