Home
last modified time | relevance | path

Searched defs:ModuleInput (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dthroughput_benchmark.h100 struct C10_HIDDEN ModuleInput { struct
104 ModuleInput& operator=(ModuleInput& other) = delete; argument
105 ModuleInput& operator=(ModuleInput&& other) = delete; argument
107 ModuleInput(py::args&& args, py::kwargs&& kwargs) in ModuleInput() argument
127 typedef BenchmarkHelper<ModuleInput, py::object, py::object> ModuleBenchmark; argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_modules.py161 class ModuleInput: class