Home
last modified time | relevance | path

Searched refs:ModuleBenchmark (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dthroughput_benchmark.h127 typedef BenchmarkHelper<ModuleInput, py::object, py::object> ModuleBenchmark; typedef
141 void ModuleBenchmark::runOnce(ModuleInput&& input) const;
144 ModuleOutput ModuleBenchmark::runOnce(
154 void ModuleBenchmark::addInput(py::args&& args, py::kwargs&& kwargs);
194 detail::ModuleBenchmark module_;
H A Dthroughput_benchmark.cpp88 void ModuleBenchmark::runOnce(ModuleInput&& input) const { in runOnce()
95 ModuleOutput ModuleBenchmark::runOnce( in runOnce()
120 void ModuleBenchmark::addInput(py::args&& args, py::kwargs&& kwargs) { in addInput()