Searched refs:GetMaxImage2DWidth (Results 1 – 4 of 4) sorted by relevance
56 return shape.w * shape.b <= gpu_info.GetMaxImage2DWidth() && in CanCreateTensorWithShape()60 return shape.w * shape.b * shape.d <= gpu_info.GetMaxImage2DWidth() && in CanCreateTensorWithShape()65 shape.w * shape.b * shape.d <= gpu_info.GetMaxImage2DWidth() && in CanCreateTensorWithShape()
355 uint64_t GetMaxImage2DWidth() const;
591 uint64_t GpuInfo::GetMaxImage2DWidth() const { in GetMaxImage2DWidth() function in tflite::gpu::GpuInfo
64 int2(gpu_info.GetMaxImage2DWidth(), gpu_info.GetMaxImage2DHeight()); in ExceedsMaxSize()