/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
D | winograd.h | 35 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,
|
D | winograd.cc | 468 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()
|
D | mean_stddev_normalization.h | 35 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
|
D | softmax1x1.h | 30 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
|
D | reduce.h | 37 TuningType tuning_type, const GpuInfo& gpu_info,
|
D | convolution_transposed_3x3.h | 41 TuningType tuning_type, const GpuInfo& gpu_info,
|
D | convolution_transposed_4x4.h | 41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
|
D | conv_metal.h | 58 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
|
D | depthwise_conv_3x3_stride_h2.h | 39 TuningType tuning_type, const GpuInfo& gpu_info,
|
D | depthwise_conv_3x3.h | 40 TuningType tuning_type, const GpuInfo& gpu_info,
|
D | fully_connected.h | 120 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
|
D | depthwise_conv_3x3_stride_h2.cc | 198 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
|
D | conv_buffer_1x1.h | 46 TuningType tuning_type, const GpuInfo& gpu_info,
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
D | work_group_picking.cc | 271 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()
|
D | work_group_picking.h | 50 void GetPossibleWorkGroups(TuningType tuning_type, const GpuInfo& gpu_info, 54 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info,
|
D | tuning_type.h | 22 enum class TuningType { kExhaustive, kFast }; enum
|
D | gpu_operation.h | 114 TuningType tuning_type, const GpuInfo& gpu_info,
|
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
D | compute_task.h | 68 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
|
D | inference_context.h | 149 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
|
D | compute_task.cc | 242 absl::Status ComputeTask::Tune(TuningType tuning_type, MetalDevice* device) { in Tune()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
D | cl_operation.h | 74 absl::Status Tune(TuningType tuning_type, const GpuInfo& gpu_info,
|
D | inference_context.cc | 169 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()
|
D | cl_operation.cc | 208 absl::Status ClOperation::Tune(TuningType tuning_type, const GpuInfo& gpu_info, in Tune()
|
D | inference_context.h | 138 absl::Status Tune(TuningType tuning_type, const GpuInfo& gpu_info,
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
D | fc_fc_add.h | 96 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups()
|