Searched refs:LocalClientOptions (Results 1 – 14 of 14) sorted by relevance
27 LocalClientOptions::LocalClientOptions( in LocalClientOptions() function in xla::LocalClientOptions36 LocalClientOptions& LocalClientOptions::set_platform(se::Platform* platform) { in set_platform()41 se::Platform* LocalClientOptions::platform() const { return platform_; } in platform()43 LocalClientOptions& LocalClientOptions::set_number_of_replicas( in set_number_of_replicas()49 int LocalClientOptions::number_of_replicas() const { in number_of_replicas()53 LocalClientOptions& LocalClientOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads()59 int LocalClientOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads()63 LocalClientOptions& LocalClientOptions::set_allowed_devices( in set_allowed_devices()69 const absl::optional<std::set<int>>& LocalClientOptions::allowed_devices() in allowed_devices()84 LocalClientOptions default_options; in GetOrCreateLocalClient()[all …]
46 class LocalClientOptions {48 LocalClientOptions(54 LocalClientOptions& set_platform(se::Platform* platform);59 LocalClientOptions& set_number_of_replicas(int number_of_replicas);63 LocalClientOptions& set_intra_op_parallelism_threads(int num_threads);68 LocalClientOptions& set_allowed_devices(92 const LocalClientOptions& options);
45 const LocalClientOptions& client_options) { in GetOrCreateLocalClientOrDie()62 se::Platform* platform, const LocalClientOptions& client_options) in ClientLibraryTestBase()67 LocalClientOptions ref_options; in ClientLibraryTestBase()86 LocalClientOptions default_options; in ClientLibraryTestBase()90 LocalClientOptions ref_options; in ClientLibraryTestBase()
61 xla::LocalClientOptions client_options; in TestWithDeviceCount()
72 const LocalClientOptions& client_options);
835 xla::LocalClientOptions client_options; in BM_ParallelFusion()
1818 xla::LocalClientOptions client_options; in DOT_ReorderContracting()
39 LocalClientOptions options; in GetInterpreterClient()
38 LocalClientOptions options; in GetCpuClient()
201 LocalClientOptions options; in GetTpuClient()
69 LocalClientOptions options; in GetGpuXlaClient()
59 xla::LocalClientOptions client_options; in BuildXlaCompilationCache()
245 xla::LocalClientOptions options; in GetOrCreateClient()
84 LocalClientOptions options; in GetCpuClientWithNonLocalDevice()