Searched refs:py_call (Results 1 – 3 of 3) sorted by relevance
81 [&](ExtraFields<EventType::PyCall>& py_call) { in calculateUniqueTensorIDs() argument83 if (py_call.module_.has_value() && in calculateUniqueTensorIDs()84 seen_modules.insert(py_call.module_->self_).second) { in calculateUniqueTensorIDs()85 for (auto& p : py_call.module_->parameters_) { in calculateUniqueTensorIDs()92 if (py_call.optimizer_.has_value() && in calculateUniqueTensorIDs()93 seen_optimizers.insert(py_call.optimizer_->self_).second) { in calculateUniqueTensorIDs()94 for (auto& p : py_call.optimizer_->parameters_) { in calculateUniqueTensorIDs()
1051 void operator()(ExtraFields<EventType::PyCall>& py_call) { in operator ()()1052 py_call.id_ = ++current_python_id_; in operator ()()1053 if (py_call.module_.has_value()) { in operator ()()1054 auto& m = py_call.module_; in operator ()()1060 void operator()(ExtraFields<EventType::PyCCall>& py_call) { in operator ()()1061 py_call.id_ = ++current_python_id_; in operator ()()
220 void operator()(const ExtraFields<EventType::PyCall>& py_call) { in operator ()()221 if (py_call.module_.has_value()) { in operator ()()222 addMetadata("Python module id", std::to_string(py_call.module_->id_)); in operator ()()