Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dthroughput_benchmark.cpp70 void ScriptModuleBenchmark::runOnce(ScriptModuleInput&& input) const { in runOnce()
82 ScriptModuleInput stack = jit::createStackForSchema( in runOnce()
114 void ScriptModuleBenchmark::addInput(ScriptModuleInput&& input) { in addInput()
133 ScriptModuleInput cloneInput<ScriptModuleInput>( in cloneInput()
134 const ScriptModuleInput& input) { in cloneInput()
H A Dthroughput_benchmark.h114 typedef std::vector<at::IValue> ScriptModuleInput; typedef
120 typedef BenchmarkHelper<ScriptModuleInput, at::IValue, jit::Module>
123 inline BenchmarkHelper<ScriptModuleInput, at::IValue, jit::Module>::
133 void ScriptModuleBenchmark::runOnce(ScriptModuleInput&& input) const;
151 void ScriptModuleBenchmark::addInput(ScriptModuleInput&& input);