Home
last modified time | relevance | path

Searched defs:op_kwargs (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/
H A Dnested.py139 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 Dtest_utils.py23 op_kwargs=None): argument
/aosp_15_r20/external/executorch/backends/arm/test/quantizer/
H A Dtest_generic_annotater.py17 def __init__(self, op, example_input, **op_kwargs) -> None:
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Dcore.py2226 op_info, device, dtype, requires_grad, op_kwargs=None, **kwargs argument