Home
last modified time | relevance | path

Searched refs:AssignComputationDevice (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h61 Status AssignComputationDevice(HloComputation* computation, int64 device);
Dhlo_sharding_util.cc213 Status AssignComputationDevice(HloComputation* computation, int64 device) { in AssignComputationDevice() function