Searched refs:GetMaxImage2DArrayLayers (Results 1 – 4 of 4) sorted by relevance
58 slices * shape.d <= gpu_info.GetMaxImage2DArrayLayers(); in CanCreateTensorWithShape()
357 uint64_t GetMaxImage2DArrayLayers() const;
617 uint64_t GpuInfo::GetMaxImage2DArrayLayers() const { in GetMaxImage2DArrayLayers() function in tflite::gpu::GpuInfo
65 size_checker.max_z_size = gpu_info.GetMaxImage2DArrayLayers(); in ExceedsMaxSize()