Searched refs:placer_ (Results 1 – 4 of 4) sorted by relevance
30 placer_ = tensorflow::tpu::ExecutorApiFn()->TpuComputationPlacer_NewFn(); in TpuComputationPlacer()34 tensorflow::tpu::ExecutorApiFn()->TpuComputationPlacer_FreeFn(placer_); in ~TpuComputationPlacer()48 placer_, replica_count, computation_count, result.data(), in AssignDevices()
46 XLA_ComputationPlacer* placer_;
392 placer_(std::move(placer)) { in SchedulerState()393 DCHECK(placer_); // check if the pointer is valid. in SchedulerState()638 return placer_->get_canonical_device_name(*node); in DeviceName()644 return absl::StrReplaceAll(placer_->get_canonical_device_name(*node), in SanitizedDeviceName()748 device = placer_->get_device(*node); in CreateOpContext()
445 std::unique_ptr<VirtualPlacer> placer_; variable