Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_placer.h77 virtual StatusOr<DeviceAssignment> AssignDevices(int replica_count,
Dcomputation_placer.cc100 StatusOr<DeviceAssignment> ComputationPlacer::AssignDevices( in AssignDevices() function in xla::ComputationPlacer
Dhlo_runner.cc421 backend().computation_placer()->AssignDevices(options.num_replicas, 1)); in ExecuteReplicated()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc357 client.client()->backend().computation_placer()->AssignDevices( in Compile()