Searched refs:constructor_input (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_modules.py | 165 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 D | test_modules.py | 62 … 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 D | test_expanded_weights.py | 823 module_input.constructor_input.args, 824 module_input.constructor_input.kwargs,
|
H A D | test_decomp.py | 688 module_input.constructor_input.args, 689 module_input.constructor_input.kwargs,
|
/aosp_15_r20/external/pytorch/test/functorch/ |
H A D | test_aotdispatch.py | 6202 module_input.constructor_input.args, 6203 module_input.constructor_input.kwargs,
|