Searched defs:TpuMemoryLimit (Results 1 – 1 of 1) sorted by relevance
114 int64 TpuPlatform::TpuMemoryLimit() { in TpuMemoryLimit() function in tensorflow::tpu::TpuPlatform168 Status TpuPlatform::TpuMemoryLimit(int64* memory_limit) { in TpuMemoryLimit() function in tensorflow::tpu::TpuPlatform