Searched refs:service_options (Results 1 – 5 of 5) sorted by relevance
35 DistributedRuntimeServiceImpl::Options service_options; in TEST() local36 service_options.num_nodes = num_nodes; in TEST()37 DistributedRuntimeServiceImpl service(service_options); in TEST()106 DistributedRuntimeServiceImpl::Options service_options; in TEST() local107 service_options.num_nodes = 2; in TEST()108 DistributedRuntimeServiceImpl service(service_options); in TEST()188 DistributedRuntimeServiceImpl::Options service_options; in TEST() local189 service_options.num_nodes = num_nodes; in TEST()190 service_options.heartbeat_interval = absl::Milliseconds(500); in TEST()191 service_options.max_missing_heartbeats = 2; in TEST()[all …]
106 ServiceOptions service_options; in GetOrCreateLocalClient() local107 service_options.set_platform(platform); in GetOrCreateLocalClient()108 service_options.set_number_of_replicas(replica_count); in GetOrCreateLocalClient()109 service_options.set_intra_op_parallelism_threads( in GetOrCreateLocalClient()111 service_options.set_allowed_devices(options.allowed_devices()); in GetOrCreateLocalClient()114 LocalService::NewService(service_options)); in GetOrCreateLocalClient()
167 ServiceExecutableRunOptions service_options(run_options, in RunHelper() local169 return std::make_pair(service_options, std::move(stream)); in RunHelper()
168 service_options = service_descriptor.GetOptions()170 self.assertEqual(-9876543210, service_options.Extensions[service_opt1])
177 service_options = service_descriptor.GetOptions()179 self.assertEqual(-9876543210, service_options.Extensions[service_opt1])