Home
last modified time | relevance | path

Searched defs:onEachDevice (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/profiler/stubs/
H A Ditt.cpp34 void onEachDevice(std::function<void(int)> op) const override {} in onEachDevice() function
H A Dbase.cpp36 void onEachDevice(std::function<void(int)>) const override { in onEachDevice() function
H A Dcuda.cpp88 void onEachDevice(std::function<void(int)> op) const override { in onEachDevice() function