Searched refs:PyCall (Results 1 – 12 of 12) sorted by relevance
| /external/pytorch/torch/csrc/autograd/ |
| D | profiler_python.cpp | 37 enum CallType { PyCall = 0, PyModuleCall, PyCCall, PyOptimizerCall }; enumerator
116 CallType::PyCall == 0,
195 struct Config<CallType::PyCall> {
199 static constexpr EventType event_type = EventType::PyCall;
221 static constexpr EventType event_type = EventType::PyCall;
265 Config<CallType::PyCall>::key_t caller_;
271 using PyCallKey = Config<CallType::PyCall>::key_t;
286 auto caller = load<CallType::PyCall>(callsite.caller_); in load()
321 value_cache->store<CallType::PyCall>(*cache.location_, no_ephemeral_t()); in set_class()
363 void ValueCache::store<CallType::PyCall>(const PyCallKey& key, no_ephemeral_t) { in store()
[all …]
|
| D | profiler_kineto.cpp | 220 void operator()(const ExtraFields<EventType::PyCall>& py_call) { in operator ()()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 63 struct PyCall { struct
88 Status MakeArgTuple(const PyCall* call, TFE_Context* ctx, PyObject** tuple) { in MakeArgTuple()
186 Status DoCallPyFunc(PyCall* call, bool* out_log_on_error) { in DoCallPyFunc()
327 PyCall call; in Compute()
|
| /external/pytorch/torch/csrc/profiler/python/ |
| D | init.cpp | 421 .value("PyCall", EventType::PyCall) in initPythonBindings()
524 py::class_<ExtraFields<EventType::PyCall>>(m, "_ExtraFields_PyCall") in initPythonBindings()
525 .def_readonly("callsite", &ExtraFields<EventType::PyCall>::callsite_) in initPythonBindings()
526 .def_readonly("caller", &ExtraFields<EventType::PyCall>::caller_) in initPythonBindings()
527 .def_readonly("module", &ExtraFields<EventType::PyCall>::module_) in initPythonBindings()
528 .def_readonly("optimizer", &ExtraFields<EventType::PyCall>::optimizer_); in initPythonBindings()
531 .def_readonly("caller", &ExtraFields<EventType::PyCall>::caller_); in initPythonBindings()
|
| /external/pytorch/torch/_C/ |
| D | _profiler.pyi | 51 PyCall = ...
101 | tuple[Literal[_EventType.PyCall], _ExtraFields_PyCall]
|
| /external/pytorch/test/profiler/ |
| D | test_memory_profiler.py | 197 self.gradient_detected(prof, _EventType.PyCall, p.grad, p),
211 self.assertGradientDetected(name, prof, _EventType.PyCall, p.grad, p)
237 self.assertGradientDetected("w0", prof, _EventType.PyCall, w0.grad, w0)
238 self.assertGradientDetected("w1", prof, _EventType.PyCall, w1.grad, w1)
256 self.gradient_detected(prof, _EventType.PyCall, w0.grad, w0),
261 self.gradient_detected(prof, _EventType.PyCall, w1.grad, w1),
289 "weight", prof, _EventType.PyCall, model[0].weight.grad, model[0].weight
|
| /external/pytorch/torch/csrc/profiler/ |
| D | collection.h | 35 PyCall, enumerator
306 struct ExtraFields<EventType::PyCall> : public PyExtraFieldsBase {
411 ExtraFields<EventType::PyCall>,
|
| D | data_flow.cpp | 81 [&](ExtraFields<EventType::PyCall>& py_call) { in calculateUniqueTensorIDs()
|
| D | collection.cpp | 541 std::string toString(const ExtraFields<EventType::PyCall>& e) { in toString()
594 ATTRIBUTE(PyCall, toString(e)), in name()
606 ATTRIBUTE(PyCall, libkineto::ActivityType::PYTHON_FUNCTION), in kinetoType()
|
| /external/pytorch/torch/profiler/ |
| D | _memory_profiler.py | 182 elif node.typed[0] == _EventType.PyCall:
348 elif node.typed[0] == _EventType.PyCall:
968 if event.typed[0] == _EventType.PyCall and event.typed[1].optimizer:
|
| D | _pattern_matcher.py | 587 if event.tag == _EventType.PyCall or event.tag == _EventType.PyCCall:
|
| /external/pytorch/torch/cuda/ |
| D | _memory_viz.py | 427 if parent.tag in (_EventType.PyCall, _EventType.PyCCall):
|