Searched refs:jax_enable_x64 (Results 1 – 2 of 2) sorted by relevance
382 bool jax_enable_x64) { in ArgSignatureOfValue() argument398 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue() argument399 if (jax_enable_x64) { in ArgSignatureOfValue()407 bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue() argument410 if (jax_enable_x64) { in ArgSignatureOfValue()418 bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue() argument423 if (jax_enable_x64) { in ArgSignatureOfValue()438 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue() argument447 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue() argument459 [](py::handle h, bool jax_enable_x64) -> xla::StatusOr<ArgSignature> { in ArgSignatureOfValue() argument[all …]
157 bool jax_enable_x64);170 bool jax_enable_x64,