Searched refs:guarded_code (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/torch/csrc/dynamo/ |
| D | cache_entry.cpp | 7 CacheEntry::CacheEntry(const py::handle& guarded_code, PyObject* backend) in CacheEntry() argument
9 this->check_fn = guarded_code.attr("check_fn"); in CacheEntry()
10 this->code = guarded_code.attr("code"); in CacheEntry()
11 this->compile_id = guarded_code.attr("compile_id"); in CacheEntry()
|
| D | extra_state.cpp | 158 PyObject* guarded_code, in create_cache_entry() argument
160 extra_state->cache_entry_list.emplace_front(guarded_code, backend); in create_cache_entry()
166 py::handle check_fn = py::handle(guarded_code).attr("check_fn"); in create_cache_entry()
|
| D | cache_entry.h | 55 CacheEntry(const py::handle& guarded_code, PyObject* backend);
|
| /external/pytorch/torch/_dynamo/ |
| D | mutation_guard.py | 33 def track(self, guarded_code): argument
34 self.watchers.append(weakref.ref(guarded_code))
37 def watch(obj, guarded_code): argument
44 tracker.track(guarded_code)
|
| D | convert_frame.py | 801 guarded_code = GuardedCode(out_code, check_fn.check_fn, compile_id)
811 return guarded_code
922 guarded_code = None
924 guarded_code = compile_inner(code, one_graph, hooks, transform)
925 return guarded_code
1035 guarded_code is not None,
|