Home
last modified time | relevance | path

Searched refs:PythonSignature (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpmap_lib.cc157 py::object PythonSignature() { in PythonSignature() function in jax::PmapFunction
420 cfun.def_property_readonly("__signature__", &PmapFunction::PythonSignature); in BuildPmapSubmodule()
Djax_jit.cc858 py::object PythonSignature() { in PythonSignature() function in jax::__anone04173081411::CompiledFunction
1216 &CompiledFunction::PythonSignature); in BuildJaxjitSubmodule()