Searched refs:GetMaxBufferSize (Results 1 – 5 of 5) sorted by relevance
526 offset_assignment.total_size <= device->GetInfo().GetMaxBufferSize()) { in AllocateMemoryForBuffers()541 if (bufferSize > device->GetInfo().GetMaxBufferSize()) { in AllocateMemoryForBuffers()546 std::to_string(device->GetInfo().GetMaxBufferSize()); in AllocateMemoryForBuffers()
472 uint64_t GetMaxBufferSize() const;
1010 uint64_t GpuInfo::GetMaxBufferSize() const { in GetMaxBufferSize() function in tflite::gpu::GpuInfo
1411 if (allocation_size > gpu_info.GetMaxBufferSize()) { in CanCreateTensorWithShape()1415 gpu_info.GetMaxBufferSize(), " bytes. ", common_desc)); in CanCreateTensorWithShape()1428 } else if (allocation_size > gpu_info.GetMaxBufferSize()) { in CanCreateTensorWithShape()1432 gpu_info.GetMaxBufferSize(), " bytes. ", common_desc)); in CanCreateTensorWithShape()
212 offset_assignment->total_size <= gpu_info.GetMaxBufferSize()) { in GetBufferAsignment()520 if (is_sub_buffers_supported && total_size <= gpu_info.GetMaxBufferSize()) { in AllocateBufferBasedTensors()