Searched refs:ModuleInput (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_modules.py | 161 class ModuleInput: class
272 ModuleInput(constructor_input=FunctionInput(10, 8),
275 ModuleInput(constructor_input=FunctionInput(10, 8, bias=False),
279 ModuleInput(constructor_input=FunctionInput(3, 5),
301 ModuleInput(constructor_input=FunctionInput(2, 3, 4),
304 ModuleInput(constructor_input=FunctionInput(2, 3, 4, bias=False),
308 ModuleInput(constructor_input=FunctionInput(2, 3, 4),
336 ModuleInput(constructor_input=FunctionInput(**constructor_kwargs),
345 ModuleInput(constructor_input=FunctionInput(**constructor_kwargs),
381 ModuleInput(constructor_input=FunctionInput(**constructor_kwargs),
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | throughput_benchmark.h | 100 struct C10_HIDDEN ModuleInput { struct
101 ModuleInput(ModuleInput&& other) = default;
103 ModuleInput(const ModuleInput&) = delete;
104 ModuleInput& operator=(ModuleInput& other) = delete; argument
105 ModuleInput& operator=(ModuleInput&& other) = delete; argument
107 ModuleInput(py::args&& args, py::kwargs&& kwargs) in ModuleInput() function
127 typedef BenchmarkHelper<ModuleInput, py::object, py::object> ModuleBenchmark; argument
129 inline BenchmarkHelper<ModuleInput, py::object, py::object>::BenchmarkHelper() in BenchmarkHelper()
141 void ModuleBenchmark::runOnce(ModuleInput&& input) const;
|
| H A D | throughput_benchmark.cpp | 88 void ModuleBenchmark::runOnce(ModuleInput&& input) const { in runOnce()
125 ModuleInput cloneInput<ModuleInput>(const ModuleInput& input) { in cloneInput()
|