Searched refs:TpuComputationPlacer (Results 1 – 3 of 3) sorted by relevance
27 using StatusOr = TpuComputationPlacer::StatusOr<T>;29 TpuComputationPlacer::TpuComputationPlacer() { in TpuComputationPlacer() function in tensorflow::tpu::TpuComputationPlacer33 TpuComputationPlacer::~TpuComputationPlacer() { in ~TpuComputationPlacer()37 StatusOr<int> TpuComputationPlacer::DeviceId(int replica, int computation, in DeviceId()43 StatusOr<xla::DeviceAssignment> TpuComputationPlacer::AssignDevices( in AssignDevices()57 TpuComputationPlacer::AssignLocalDevices(TpuHostLocationExternal host_location, in AssignLocalDevices()72 return std::make_unique<TpuComputationPlacer>(); in CreateTpuComputationPlacer()
27 class TpuComputationPlacer : public xla::ComputationPlacer {32 TpuComputationPlacer();33 ~TpuComputationPlacer() override;
107 return tf_tpu::TpuComputationPlacer::AssignLocalDevices(host, num_replicas, in GetDefaultDeviceAssignment()