Home
last modified time | relevance | path

Searched refs:TuningType (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dwinograd.h35 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
69 TuningType tuning_type, const GpuInfo& gpu_info,
102 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
135 TuningType tuning_type, const GpuInfo& gpu_info,
Dwinograd.cc468 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
475 case TuningType::kExhaustive: in GetPossibleKernelWorkGroups()
479 case TuningType::kFast: in GetPossibleKernelWorkGroups()
682 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
689 case TuningType::kExhaustive: in GetPossibleKernelWorkGroups()
693 case TuningType::kFast: in GetPossibleKernelWorkGroups()
Dmean_stddev_normalization.h35 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dsoftmax1x1.h30 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dreduce.h37 TuningType tuning_type, const GpuInfo& gpu_info,
Dconvolution_transposed_3x3.h41 TuningType tuning_type, const GpuInfo& gpu_info,
Dconvolution_transposed_4x4.h41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dconv_metal.h58 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Ddepthwise_conv_3x3_stride_h2.h39 TuningType tuning_type, const GpuInfo& gpu_info,
Ddepthwise_conv_3x3.h40 TuningType tuning_type, const GpuInfo& gpu_info,
Dfully_connected.h120 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Ddepthwise_conv_3x3_stride_h2.cc198 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
Dconv_buffer_1x1.h46 TuningType tuning_type, const GpuInfo& gpu_info,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dwork_group_picking.cc271 void GetPossibleWorkGroups(TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleWorkGroups()
279 case TuningType::kFast: in GetPossibleWorkGroups()
283 case TuningType::kExhaustive: { in GetPossibleWorkGroups()
293 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleWorkGroupsConv()
301 case TuningType::kFast: { in GetPossibleWorkGroupsConv()
311 case TuningType::kExhaustive: { in GetPossibleWorkGroupsConv()
Dwork_group_picking.h50 void GetPossibleWorkGroups(TuningType tuning_type, const GpuInfo& gpu_info,
54 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info,
Dtuning_type.h22 enum class TuningType { kExhaustive, kFast }; enum
Dgpu_operation.h114 TuningType tuning_type, const GpuInfo& gpu_info,
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.h68 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
Dinference_context.h149 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
Dcompute_task.cc242 absl::Status ComputeTask::Tune(TuningType tuning_type, MetalDevice* device) { in Tune()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.h74 absl::Status Tune(TuningType tuning_type, const GpuInfo& gpu_info,
Dinference_context.cc169 TuningType tuning_type = TuningType::kExhaustive; in InitFromGraph()
171 tuning_type = TuningType::kFast; in InitFromGraph()
178 tuning_type = TuningType::kFast; in InitFromGraph()
634 absl::Status InferenceContext::Tune(TuningType tuning_type, in Tune()
Dcl_operation.cc208 absl::Status ClOperation::Tune(TuningType tuning_type, const GpuInfo& gpu_info, in Tune()
Dinference_context.h138 absl::Status Tune(TuningType tuning_type, const GpuInfo& gpu_info,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Dfc_fc_add.h96 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()

12