Home
last modified time | relevance | path

Searched refs:ComputationPlacer (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_placer.cc124 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 …]
Dcomputation_placer.h73 class ComputationPlacer {
75 ComputationPlacer() {} in ComputationPlacer() function
76 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);
Dbackend.h95 ComputationPlacer* computation_placer() const { return computation_placer_; } in computation_placer()
162 ComputationPlacer* computation_placer,
170 ComputationPlacer* computation_placer_;
Dtpu_computation_placer.cc71 static std::unique_ptr<xla::ComputationPlacer> CreateTpuComputationPlacer() { in CreateTpuComputationPlacer()
76 xla::ComputationPlacer::RegisterComputationPlacer(GetTpuPlatformId(), in InitModule()
Dbackend.cc92 ComputationPlacer::GetForPlatform(platform)); in CreateBackend()
124 ComputationPlacer* computation_placer, in Backend()
Dtpu_computation_placer.h27 class TpuComputationPlacer : public xla::ComputationPlacer {
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc175 xla::ComputationPlacer::RegisterComputationPlacer( in InitModule()
177 []() { return absl::make_unique<xla::ComputationPlacer>(); }); in InitModule()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc149 xla::ComputationPlacer placer; in GetDefaultDeviceAssignment()