| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | winograd.h | 35 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
69 TuningType tuning_type, const GpuInfo& gpu_info,
102 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
135 TuningType tuning_type, const GpuInfo& gpu_info,
|
| D | winograd.cc | 468 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
474 switch (tuning_type) { in GetPossibleKernelWorkGroups()
476 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
682 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
688 switch (tuning_type) { in GetPossibleKernelWorkGroups()
690 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | depthwise_conv_3x3_stride_h2.cc | 198 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
203 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | mean_stddev_normalization.h | 35 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
|
| D | softmax1x1.h | 30 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
|
| 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() argument
|
| D | depthwise_conv_3x3.cc | 292 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
297 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | conv_metal.h | 58 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
|
| D | convolution_transposed_3x3.cc | 353 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
360 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, 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() argument
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| 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 | BUILD | 73 "//tensorflow/lite/delegates/gpu/common/task:tuning_type",
174 name = "tuning_type",
175 hdrs = ["tuning_type.h"],
219 ":tuning_type",
|
| D | work_group_picking.cc | 271 void GetPossibleWorkGroups(TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleWorkGroups() argument
278 switch (tuning_type) { in GetPossibleWorkGroups()
293 void GetPossibleWorkGroupsConv(TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleWorkGroupsConv() argument
300 switch (tuning_type) { in GetPossibleWorkGroupsConv()
|
| D | gpu_operation.cc | 210 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
212 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| 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 | compute_task.cc | 242 absl::Status ComputeTask::Tune(TuningType tuning_type, MetalDevice* device) { in Tune() argument
247 operation_->GetPossibleKernelWorkGroups(tuning_type, device->GetInfo(), in Tune()
|
| D | BUILD | 124 "//tensorflow/lite/delegates/gpu/common/task:tuning_type",
167 "//tensorflow/lite/delegates/gpu/common/task:tuning_type",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.cc | 169 TuningType tuning_type = TuningType::kExhaustive; in InitFromGraph() local
171 tuning_type = TuningType::kFast; in InitFromGraph()
178 tuning_type = TuningType::kFast; in InitFromGraph()
182 Tune(tuning_type, env->device().GetInfo(), env->profiling_queue())); in InitFromGraph()
634 absl::Status InferenceContext::Tune(TuningType tuning_type, in Tune() argument
639 node.cl_operation.Tune(tuning_type, gpu_info, profiling_queue)); in Tune()
|
| D | cl_operation.cc | 208 absl::Status ClOperation::Tune(TuningType tuning_type, const GpuInfo& gpu_info, in Tune() argument
211 operation_->GetPossibleKernelWorkGroups(tuning_type, gpu_info, kernel_.info_, in Tune()
|
| D | cl_operation.h | 74 absl::Status Tune(TuningType tuning_type, const GpuInfo& gpu_info,
|