Home
last modified time | relevance | path

Searched refs:IsApiVulkan (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_info.cc466 } else if (IsApiVulkan()) { in SupportsExtension()
528 if (IsApiVulkan()) { in GetMaxWorkGroupSizeForX()
544 if (IsApiVulkan()) { in GetMaxWorkGroupSizeForY()
560 if (IsApiVulkan()) { in GetMaxWorkGroupSizeForZ()
576 if (IsApiVulkan()) { in GetMaxWorkGroupTotalSize()
595 if (IsApiVulkan()) { in GetMaxImage2DWidth()
608 if (IsApiVulkan()) { in GetMaxImage2DHeight()
621 if (IsApiVulkan()) { in GetMaxImage2DArrayLayers()
671 if (IsApiVulkan()) { in GetMaxImageArguments()
693 bool GpuInfo::IsApiVulkan() const { return gpu_api == GpuApi::kVulkan; } in IsApiVulkan() function in tflite::gpu::GpuInfo
Dgpu_info.h380 bool IsApiVulkan() const;
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dmean.cc36 return gpu_info.IsApiVulkan() && in UseSubgroupBasedImpl()