pytorch/torch/csrc/jit/frontend
Antonio Kim 7fc292930c Add support for torch.Generator type in TorchScript (#110413)
- Add support for `torch.Generator` type in TorchScript
- Add `generator` args to all `torch.nn.init` functions that call `uniform_` or `normal_`
- Add support for `torch.Generator` in LTC's TorchScript backend (CC: @wconstab)

CC: @eellison @davidberard98 @GlebKazantaev @behzad-a
Pull Request resolved: https://github.com/pytorch/pytorch/pull/110413
Approved by: https://github.com/wconstab, https://github.com/albanD, https://github.com/glebk-cerebras, https://github.com/davidberard98
2023-11-21 23:07:21 +00:00
..
builtin_functions.cpp
builtin_functions.h
canonicalize_modified_loop.cpp
canonicalize_modified_loop.h
concrete_module_type.cpp [RELAND] Remove some unnecessary <iostream> includes from headers (#108150) 2023-09-20 21:55:15 +00:00
concrete_module_type.h
convert_to_ssa.cpp [Reland 1] Cleanup header inclusions in torch_cpu by iwyu (#112311) 2023-11-19 04:06:36 +00:00
convert_to_ssa.h
edit_distance.cpp
edit_distance.h
error_report.cpp [Reland 1] Cleanup header inclusions in torch_cpu by iwyu (#112311) 2023-11-19 04:06:36 +00:00
error_report.h
exit_transforms.cpp [Reland 1] Cleanup header inclusions in torch_cpu by iwyu (#112311) 2023-11-19 04:06:36 +00:00
exit_transforms.h
function_schema_parser.cpp [Reland2] Remove calls of c10::either (#110487) 2023-10-06 00:25:15 +00:00
function_schema_parser.h [Reland2] Remove calls of c10::either (#110487) 2023-10-06 00:25:15 +00:00
inline_loop_condition.cpp
inline_loop_condition.h
ir_emitter.cpp [Reland] Move torch::make_unique to std::make_unique (#109780) 2023-09-21 18:30:21 +00:00
ir_emitter.h
lexer.cpp [Reland] Elimates c10::guts::to_string (#108748) 2023-09-07 13:35:17 +00:00
lexer.h Revert "[1/N] Cleanup header inclusions in torch_cpu by iwyu (#101178)" 2023-09-25 20:05:25 +00:00
mini_environment.h
name_mangler.cpp [Reland] Eliminate invocations of c10::stoi,c10::stod,c10::stoull,c10::stoll (#109566) 2023-09-19 07:15:25 +00:00
name_mangler.h
parse_string_literal.h
parser_constants.h
parser.cpp
parser.h
resolver.h
schema_matching.cpp
schema_matching.h
schema_type_parser.cpp [Reland] Eliminate invocations of c10::stoi,c10::stod,c10::stoull,c10::stoll (#109566) 2023-09-19 07:15:25 +00:00
schema_type_parser.h
script_type_parser.cpp Python 3.10 Union operator | support for JIT (#109293) 2023-09-25 15:35:54 +00:00
script_type_parser.h
source_range.cpp Add regex matching to Inductor all2all collective unit tests (#112077) 2023-10-26 08:29:30 +00:00
source_range.h [2/N] Fixes clang-tidy warnings in header files (#113727) 2023-11-16 13:21:15 +00:00
source_ref.h
strtod.cpp
strtod.h
sugared_value.cpp Add support for torch.Generator type in TorchScript (#110413) 2023-11-21 23:07:21 +00:00
sugared_value.h
tracer.cpp Add support for torch.Generator type in TorchScript (#110413) 2023-11-21 23:07:21 +00:00
tracer.h [RELAND] Remove some unnecessary <iostream> includes from headers (#108150) 2023-09-20 21:55:15 +00:00
tree_views.cpp
tree_views.h Python 3.10 Union operator | support for JIT (#109293) 2023-09-25 15:35:54 +00:00
tree.h
versioned_symbols.cpp
versioned_symbols.h