Searched refs:DDPLoggingData (Results 1 – 6 of 6) sorted by relevance
26 at::DDPLoggingData get_ddp_logging_data();98 std::unique_ptr<at::DDPLoggingData> ddp_logging_data_;
61 ddp_logging_data_ = std::make_unique<at::DDPLoggingData>(); in Logger()408 at::DDPLoggingData Logger::get_ddp_logging_data() { in get_ddp_logging_data()
3045 py::class_<c10::DDPLoggingData>(module, "DDPLoggingData") in c10d_init()3047 .def_readwrite("strs_map", &c10::DDPLoggingData::strs_map) in c10d_init()3048 .def_readwrite("ints_map", &c10::DDPLoggingData::ints_map); in c10d_init()
121 using DDPUsageLoggerType = std::function<void(const DDPLoggingData&)>;148 static DDPUsageLoggerType func = [](const DDPLoggingData&) {}; in GetDDPUsageLogger() argument205 std::function<void(const DDPLoggingData&)> logger) { in SetPyTorchDDPUsageLogger()236 void LogPyTorchDDPUsage(const DDPLoggingData& ddpData) try { in LogPyTorchDDPUsage()
346 struct DDPLoggingData { struct354 std::function<void(const DDPLoggingData&)> logger); argument355 C10_API void LogPyTorchDDPUsage(const DDPLoggingData& ddpData);
76 class DDPLoggingData:93 def _get_ddp_logging_data(self) -> DDPLoggingData: ...