Searched refs:TpuMemoryLimit (Results 1 – 4 of 4) sorted by relevance
60 int64 TpuMemoryLimit() override;138 static Status TpuMemoryLimit(int64* memory_limit);
113 int64 TpuPlatform::TpuMemoryLimit() { in TpuMemoryLimit() function in tensorflow::tpu::TpuPlatform167 Status TpuPlatform::TpuMemoryLimit(int64* memory_limit) { in TpuMemoryLimit() function in tensorflow::tpu::TpuPlatform
52 virtual int64 TpuMemoryLimit() = 0;
430 TF_RETURN_IF_ERROR(tpu::TpuPlatform::TpuMemoryLimit(&memory_limit)); in CreateDevices()