Home
last modified time | relevance | path

Searched defs:AutogradCompilerCall (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/dynamo/
H A Dcompiled_autograd.h189 struct AutogradCompilerCall { struct
190 void add_size_input(const c10::SymInt& s) { in add_size_input()
195 size_t emplace_hook(c10::SafePyObject&& fn) { in emplace_hook()
200 TensorArgs tensor_args;
201 std::vector<SizeInput> all_size_inputs;
202 LiftedIValueArgs lifted_ivalue_args;
203 std::vector<int64_t> dyn_size_inputs;
204 std::vector<c10::SafePyObject> hooks;
205 NodeCalls node_calls;
206 SizeInput::DynType default_dyn_type = SizeInput::STATIC;