Searched refs:LocalClientOptions (Results 1 – 14 of 14) sorted by relevance
29 LocalClientOptions::LocalClientOptions( in LocalClientOptions() function in xla::LocalClientOptions38 LocalClientOptions& LocalClientOptions::set_platform(se::Platform* platform) { in set_platform()43 se::Platform* LocalClientOptions::platform() const { return platform_; } in platform()45 LocalClientOptions& LocalClientOptions::set_number_of_replicas( in set_number_of_replicas()51 int LocalClientOptions::number_of_replicas() const { in number_of_replicas()55 LocalClientOptions& LocalClientOptions::set_intra_op_parallelism_threads( in set_intra_op_parallelism_threads()61 int LocalClientOptions::intra_op_parallelism_threads() const { in intra_op_parallelism_threads()65 LocalClientOptions& LocalClientOptions::set_allowed_devices( in set_allowed_devices()71 const std::optional<std::set<int>>& LocalClientOptions::allowed_devices() in allowed_devices()86 LocalClientOptions default_options; in GetOrCreateLocalClient()[all …]
44 class LocalClientOptions {46 LocalClientOptions(52 LocalClientOptions& set_platform(se::Platform* platform);57 LocalClientOptions& set_number_of_replicas(int number_of_replicas);61 LocalClientOptions& set_intra_op_parallelism_threads(int num_threads);66 LocalClientOptions& set_allowed_devices(90 const LocalClientOptions& options);
44 const LocalClientOptions& client_options) { in GetOrCreateLocalClientOrDie()61 se::Platform* platform, const LocalClientOptions& client_options) in ClientLibraryTestBase()66 LocalClientOptions ref_options; in ClientLibraryTestBase()85 LocalClientOptions default_options; in ClientLibraryTestBase()89 LocalClientOptions ref_options; in ClientLibraryTestBase()
61 xla::LocalClientOptions client_options; in TestWithDeviceCount()
70 const LocalClientOptions& client_options);
833 xla::LocalClientOptions client_options; in BM_ParallelFusion()
1819 xla::LocalClientOptions client_options; in DOT_ReorderContracting()
39 LocalClientOptions options; in GetInterpreterClient()
38 LocalClientOptions options; in GetCpuClient()
286 LocalClientOptions options; in GetTpuClient()
162 LocalClientOptions options; in GetGpuXlaClient()
88 xla::LocalClientOptions client_options; in BuildXlaCompilationCache()
254 xla::LocalClientOptions options; in GetOrCreateClient()
84 LocalClientOptions options; in GetCpuClientWithNonLocalDevice()