Searched refs:set_device_assignment (Results 1 – 5 of 5) sorted by relevance
79 ExecutableRunOptions& ExecutableRunOptions::set_device_assignment( in set_device_assignment() function in xla::ExecutableRunOptions
76 ExecutableRunOptions& set_device_assignment(
250 options.set_device_assignment(&device_assignment_); in Execute()298 options.set_device_assignment(&device_assignment_); in ExecutePerReplica()
453 options.set_device_assignment(&device_assignment); in ExecuteParallelAndRegisterResult()548 options.set_device_assignment(&device_assignment); in ExecuteAndRegisterResult()
451 run_options.set_device_assignment(device_assignment); in GetServiceRunOptionsForDevice()