Searched refs:ComputationPlacer (Results 1 – 8 of 8) sorted by relevance
124 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()133 StatusOr<DeviceAssignment> ComputationPlacer::AssignDevices( in AssignDevices()147 /* static */ void ComputationPlacer::RegisterComputationPlacer( in RegisterComputationPlacer()151 ComputationPlacer::platform_computation_placer_mutex_); in RegisterComputationPlacer()157 /* static */ StatusOr<ComputationPlacer*> ComputationPlacer::GetForPlatform( in GetForPlatform()160 ComputationPlacer::platform_computation_placer_mutex_); in GetForPlatform()180 ComputationPlacer::platform_computation_placer_mutex_(183 /* static */ std::map<se::Platform::Id, ComputationPlacer::State>*184 ComputationPlacer::GetPlatformComputationPlacers() { in GetPlatformComputationPlacers()185 static auto* r = new std::map<se::Platform::Id, ComputationPlacer::State>; in GetPlatformComputationPlacers()[all …]
73 class ComputationPlacer {75 ComputationPlacer() {} in ComputationPlacer() function76 virtual ~ComputationPlacer() {} in ~ComputationPlacer()90 std::unique_ptr<ComputationPlacer> (*)();99 static StatusOr<ComputationPlacer*> GetForPlatform(110 std::unique_ptr<ComputationPlacer> placer;117 TF_DISALLOW_COPY_AND_ASSIGN(ComputationPlacer);
95 ComputationPlacer* computation_placer() const { return computation_placer_; } in computation_placer()162 ComputationPlacer* computation_placer,170 ComputationPlacer* computation_placer_;
71 static std::unique_ptr<xla::ComputationPlacer> CreateTpuComputationPlacer() { in CreateTpuComputationPlacer()76 xla::ComputationPlacer::RegisterComputationPlacer(GetTpuPlatformId(), in InitModule()
92 ComputationPlacer::GetForPlatform(platform)); in CreateBackend()124 ComputationPlacer* computation_placer, in Backend()
27 class TpuComputationPlacer : public xla::ComputationPlacer {
175 xla::ComputationPlacer::RegisterComputationPlacer( in InitModule()177 []() { return absl::make_unique<xla::ComputationPlacer>(); }); in InitModule()
149 xla::ComputationPlacer placer; in GetDefaultDeviceAssignment()