Searched refs:python_tid (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/autograd/ |
| D | profiler_python.cpp | 285 auto load(const Callsite<C>& callsite, size_t python_tid) const { in load()
290 python_tid, in load()
919 for (size_t python_tid : c10::irange(tls.size())) { local
921 tls[python_tid].trace_keys_, *this, value_cache, python_tid);
923 addExits<EventType::PyCall>(tls[python_tid].exit_times_, python_tid);
924 addExits<EventType::PyCCall>(tls[python_tid].c_exit_times_, python_tid);
944 size_t python_tid) { in operator ()() argument
947 {it.second, value_cache.load(it.first, python_tid)}); in operator ()()
955 size_t python_tid) { in addExits() argument
957 get_state<E>().exits_.push({time_converter_(i), python_tid}); in addExits()
[all …]
|
| /external/pytorch/torch/csrc/profiler/ |
| D | collection.h | 291 size_t python_tid,
294 python_tid_{python_tid},
315 size_t python_tid,
318 : PyExtraFieldsBase(end_time_ns, python_tid, std::move(caller)),
334 size_t python_tid,
337 : PyExtraFieldsBase(end_time_ns, python_tid, std::move(caller)),
|