Home
last modified time | relevance | path

Searched defs:tuning_type (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dwinograd.h36 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
106 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dmean_stddev_normalization.h41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dsoftmax1x1.h32 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dconvolution_transposed_4x4.h41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dconv_metal_simd.h33 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Ddepthwise_conv_3x3_stride_h2.cc204 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dfully_connected.h121 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Ddepthwise_conv_3x3.cc306 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dwinograd.cc514 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
732 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dconvolution_transposed_3x3.cc347 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Ddepthwise_conv.cc131 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dreduce.cc513 void Reduce::GetPossibleKernelWorkGroups(TuningType tuning_type, in GetPossibleKernelWorkGroups()
Dconvolution_transposed.cc602 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dconv_generic.cc362 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc191 absl::Status ClOperation::Tune(TuningType tuning_type, const GpuInfo& gpu_info, in Tune()
Dinference_context.cc286 TuningType tuning_type = TuningType::kExhaustive; in InitFromGpuModel() local
682 absl::Status InferenceContext::Tune(TuningType tuning_type, in Tune()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dwork_group_picking.cc271 void GetPossibleWorkGroups(TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleWorkGroups()
293 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleWorkGroupsConv()
Dgpu_operation.cc438 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleDispatches()
455 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h100 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc340 absl::Status ComputeTask::Tune(TuningType tuning_type, MetalDevice* device) { in Tune()
Dinference_context.cc649 absl::Status InferenceContext::Tune(TuningType tuning_type, in Tune()