Searched refs:GetMaxMemoryAllocationSize (Results 1 – 3 of 3) sorted by relevance
41 if (allocation_size > gpu_info.GetMaxMemoryAllocationSize()) { in CanCreateTensorWithShape()45 gpu_info.GetMaxMemoryAllocationSize(), " bytes. ", common_desc)); in CanCreateTensorWithShape()
392 uint64_t GetMaxMemoryAllocationSize() const;
697 uint64_t GpuInfo::GetMaxMemoryAllocationSize() const { in GetMaxMemoryAllocationSize() function in tflite::gpu::GpuInfo