Searched refs:ComputationPlacer (Results 1 – 5 of 5) sorted by relevance
91 StatusOr<int> ComputationPlacer::DeviceId(int replica, int computation, in DeviceId()100 StatusOr<DeviceAssignment> ComputationPlacer::AssignDevices( in AssignDevices()114 /* static */ void ComputationPlacer::RegisterComputationPlacer( in RegisterComputationPlacer()118 ComputationPlacer::platform_computation_placer_mutex_); in RegisterComputationPlacer()124 /* static */ StatusOr<ComputationPlacer*> ComputationPlacer::GetForPlatform( in GetForPlatform()127 ComputationPlacer::platform_computation_placer_mutex_); in GetForPlatform()147 ComputationPlacer::platform_computation_placer_mutex_(150 /* static */ std::map<se::Platform::Id, ComputationPlacer::State>*151 ComputationPlacer::GetPlatformComputationPlacers() { in GetPlatformComputationPlacers()152 static auto* r = new std::map<se::Platform::Id, ComputationPlacer::State>; in GetPlatformComputationPlacers()[all …]
64 class ComputationPlacer {66 ComputationPlacer() {} in ComputationPlacer() function67 virtual ~ComputationPlacer() {} in ~ComputationPlacer()81 std::unique_ptr<ComputationPlacer> (*)();90 static StatusOr<ComputationPlacer*> GetForPlatform(101 std::unique_ptr<ComputationPlacer> placer;108 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_;
114 ComputationPlacer::GetForPlatform(platform)); in CreateBackend()146 ComputationPlacer* computation_placer, in Backend()
194 xla::ComputationPlacer::RegisterComputationPlacer( in InitModule()196 []() { return absl::make_unique<xla::ComputationPlacer>(); }); in InitModule()