Searched refs:LocalClientOptions (Results 1 – 8 of 8) 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()
60 xla::LocalClientOptions client_options; in TestWithDeviceCount()
71 const LocalClientOptions& client_options);
836 xla::LocalClientOptions client_options; in BM_ParallelFusion()
235 xla::LocalClientOptions client_options; in BuildCompilationCache()
72 LocalClientOptions options; in Get()