Searched refs:TpuPlatform (Results 1 – 6 of 6) sorted by relevance
28 const ::stream_executor::Platform::Id TpuPlatform::kId = GetTpuPlatformId();29 TpuPlatform* tpu_registered_platform = nullptr;35 TpuPlatform::TpuPlatform() : name_("TPU") { in TpuPlatform() function in tensorflow::tpu::TpuPlatform39 TpuPlatform* TpuPlatform::GetRegisteredPlatform() { in GetRegisteredPlatform()43 Status TpuPlatform::Initialize( in Initialize()69 bool TpuPlatform::Initialized() const { in Initialized()73 TpuPlatform::~TpuPlatform() { in ~TpuPlatform()77 int TpuPlatform::VisibleDeviceCount() const { in VisibleDeviceCount()81 StatusOr<::stream_executor::StreamExecutor*> TpuPlatform::GetExecutor( in GetExecutor()88 TpuPlatform::GetUncachedExecutor( in GetUncachedExecutor()[all …]
33 class TpuPlatform : public ::tensorflow::tpu::TpuPlatformInterface {48 TpuPlatform();50 ~TpuPlatform() override;52 static TpuPlatform* GetRegisteredPlatform();
223 TpuPlatform& tpu_platform() { in tpu_platform()224 return *(tensorflow::down_cast<TpuPlatform*>(platform_)); in tpu_platform()227 TpuPlatform::StreamMap& stream_map() { in stream_map()
81 TpuPlatform::GetRegisteredPlatform()->stream_map()->at( in TransferLiteralToDeviceAsync()204 TpuPlatform::GetRegisteredPlatform()->LookupStream( in TransferLiteralFromDevice()287 TpuPlatform::GetRegisteredPlatform()->stream_map()->at( in WriteSingleTupleIndexTable()
74 auto platform = tensorflow::down_cast<tensorflow::tpu::TpuPlatform*>( in LoadProgramAndEnqueueToStream()
408 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpusPerHost(&device_count)); in ListPhysicalDevices()423 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpusPerHost(&device_count)); in CreateDevices()430 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpuMemoryLimit(&memory_limit)); in CreateDevices()