Searched defs:jax_enable_x64 (Results 1 – 1 of 1) sorted by relevance
382 bool jax_enable_x64) { in ArgSignatureOfValue()398 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()407 bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()418 bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()438 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()447 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()459 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()498 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()506 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()514 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue()[all …]