Home
last modified time | relevance | path

Searched refs:IsApiMetal (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dgpu_info.cc534 if (IsApiMetal()) { in GetMaxWorkGroupSizeForX()
550 if (IsApiMetal()) { in GetMaxWorkGroupSizeForY()
566 if (IsApiMetal()) { in GetMaxWorkGroupSizeForZ()
582 if (IsApiMetal()) { in GetMaxWorkGroupTotalSize()
654 } else if (IsApiMetal()) { in GetMaxBufferSize()
674 if (IsApiMetal()) { in GetMaxImageArguments()
695 bool GpuInfo::IsApiMetal() const { return gpu_api == GpuApi::kMetal; } in IsApiMetal() function in tflite::gpu::GpuInfo
Dgpu_info.h383 bool IsApiMetal() const;
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Dconvolution_selector.cc141 if (gpu_info.IsApiMetal() && IsConvolutionMetalSupported(op_def)) { in SelectConvolution()
163 if (gpu_info.IsApiMetal() && IsConvolutionMetalSupported(op_def)) { in SelectConvolutionForWinograd()
188 if (gpu_info.IsApiMetal() && IsConvolutionMetalSupported(op_def)) { in SelectConvolutionWithDynamicWeights()
Dfully_connected_selector.cc87 if (gpu_info.IsApiMetal()) { in SelectFullyConnected()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.cc347 if (gpu_info.IsApiMetal()) { in Read()
364 } else if (gpu_info.IsApiMetal()) { in Read()
380 } else if (gpu_info.IsApiMetal()) { in Read()
397 } else if (gpu_info.IsApiMetal()) { in Read()
411 } else if (gpu_info.IsApiMetal()) { in Read()
439 } else if (gpu_info.IsApiMetal()) { in Write()
450 } else if (gpu_info.IsApiMetal()) { in Write()
461 } else if (gpu_info.IsApiMetal()) { in Write()
Dtexture2d_desc.cc54 if (gpu_info.IsApiMetal()) { in PerformReadSelector()
Dtensor_linear_desc.cc82 if (gpu_info.IsApiMetal()) { in PerformReadSelector()
Dstorage_type_util.cc87 if (gpu_info.IsApiMetal()) { in SelectBestStorageType()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dmean_stddev_normalization.cc164 if (gpu_info.IsApiMetal()) { in GetNormalizationCode()
Dfully_connected_test_util.cc143 env->GetGpuInfo().IsApiMetal() && env->GetGpuInfo().IsIntel()) { in FullyConnectedExtraLargeTest()