Home
last modified time | relevance | path

Searched defs:py_kwargs (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Doptimizer.py318 self, tx: "InstructionTranslator", args, kwargs, py_args, py_kwargs argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Djax_jit.cc234 const std::optional<py::kwargs>& py_kwargs, in ParseArguments()
1161 std::optional<py::kwargs> py_kwargs; in JaxCompiledFunction_tp_call() local
H A Dpmap_lib.cc684 py::kwargs py_kwargs; in JaxPmapFunction_tp_call() local