Home
last modified time | relevance | path

Searched refs:constructor_input (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_modules.py165 def __init__(self, constructor_input, forward_input=None, desc='', reference_fn=None): argument
166 self.constructor_input = constructor_input # Inputs to pass during construction
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),
[all …]
/aosp_15_r20/external/pytorch/test/
H A Dtest_modules.py62 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
92 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
156 …args, kwargs = module_input_device.constructor_input.args, module_input_device.constructor_input.k…
205 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
224 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
228 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
272 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
380 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
452 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
566 … args, kwargs = module_input.constructor_input.args, module_input.constructor_input.kwargs
[all …]
H A Dtest_expanded_weights.py823 module_input.constructor_input.args,
824 module_input.constructor_input.kwargs,
H A Dtest_decomp.py688 module_input.constructor_input.args,
689 module_input.constructor_input.kwargs,
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_aotdispatch.py6202 module_input.constructor_input.args,
6203 module_input.constructor_input.kwargs,