Searched defs:op_kwargs (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
H A D | nested.py | 139 op_info, device, dtype, requires_grad, op_kwargs=None, **kwargs argument 151 op_info, device, dtype, requires_grad, op_kwargs=None, **kwargs argument 166 op_info, device, dtype, requires_grad, op_kwargs=None, **kwargs argument 242 op_info, device, dtype, requires_grad, op_kwargs=None, **kwargs argument
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
H A D | test_utils.py | 23 op_kwargs=None): argument
|
/aosp_15_r20/external/executorch/backends/arm/test/quantizer/ |
H A D | test_generic_annotater.py | 17 def __init__(self, op, example_input, **op_kwargs) -> None:
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
H A D | core.py | 2226 op_info, device, dtype, requires_grad, op_kwargs=None, **kwargs argument
|