Searched defs:pyclient (Results 1 – 1 of 1) sorted by relevance
556 xla::Status ComputeSignature(bool jax_enable_x64, xla::PyClient& pyclient, in ComputeSignature()1353 std::shared_ptr<xla::PyClient>& pyclient = to_device.client; in BuildJaxjitSubmodule() local