Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dprofiler_legacy.h351 struct TORCH_API RecordProfile { struct
359 std::unique_ptr<std::ofstream> file_;
360 std::ostream& out_;
H A Dprofiler_legacy.cpp643 RecordProfile::RecordProfile(std::ostream& out) : out_(out) { in RecordProfile() function in torch::autograd::profiler::RecordProfile
647 RecordProfile::RecordProfile(const std::string& filename) in RecordProfile() function in torch::autograd::profiler::RecordProfile
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dthroughput_benchmark-inl.h106 using RecordProfile = torch::autograd::profiler::RecordProfile; in benchmark() local