Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dthroughput_benchmark.h121 ScriptModuleBenchmark; typedef
133 void ScriptModuleBenchmark::runOnce(ScriptModuleInput&& input) const;
136 ScriptModuleOutput ScriptModuleBenchmark::runOnce(
149 void ScriptModuleBenchmark::addInput(py::args&& args, py::kwargs&& kwargs);
151 void ScriptModuleBenchmark::addInput(ScriptModuleInput&& input);
193 detail::ScriptModuleBenchmark script_module_;
H A Dthroughput_benchmark.cpp70 void ScriptModuleBenchmark::runOnce(ScriptModuleInput&& input) const { in runOnce()
77 ScriptModuleOutput ScriptModuleBenchmark::runOnce( in runOnce()
104 void ScriptModuleBenchmark::addInput(py::args&& args, py::kwargs&& kwargs) { in addInput()
114 void ScriptModuleBenchmark::addInput(ScriptModuleInput&& input) { in addInput()