Home
last modified time | relevance | path

Searched refs:AssignDevices (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dtpu_computation_placer.h38 StatusOr<xla::DeviceAssignment> AssignDevices(int replica_count,
Dtpu_computation_placer.cc43 StatusOr<xla::DeviceAssignment> TpuComputationPlacer::AssignDevices( in AssignDevices() function in tensorflow::tpu::TpuComputationPlacer
Dcomputation_placer.h86 virtual StatusOr<DeviceAssignment> AssignDevices(int replica_count,
Dcomputation_placer.cc133 StatusOr<DeviceAssignment> ComputationPlacer::AssignDevices( in AssignDevices() function in xla::ComputationPlacer
Dhlo_runner.cc375 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
422 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc150 return placer.AssignDevices(num_replicas, num_partitions); in GetDefaultDeviceAssignment()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc256 return client_->backend().computation_placer()->AssignDevices(num_replicas, in GetDefaultDeviceAssignment()