Searched refs:AssignLocalDevices (Results 1 – 3 of 3) sorted by relevance
41 static StatusOr<xla::DeviceAssignment> AssignLocalDevices(
57 TpuComputationPlacer::AssignLocalDevices(TpuHostLocationExternal host_location, in AssignLocalDevices() function in tensorflow::tpu::TpuComputationPlacer
127 return tf_tpu::TpuComputationPlacer::AssignLocalDevices(host, num_replicas, in GetDefaultDeviceAssignment()