| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | winograd.h | 36 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
72 TuningType tuning_type, const GpuInfo& gpu_info,
106 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
139 TuningType tuning_type, const GpuInfo& gpu_info,
|
| D | winograd.cc | 514 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
520 switch (tuning_type) { in GetPossibleKernelWorkGroups()
522 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
732 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
738 switch (tuning_type) { in GetPossibleKernelWorkGroups()
740 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | softmax1x1.h | 32 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
|
| D | depthwise_conv_3x3_stride_h2.cc | 204 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
209 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | mean_stddev_normalization.h | 41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
|
| D | reduce.h | 38 TuningType tuning_type, const GpuInfo& gpu_info,
|
| D | conv_metal_simd.h | 33 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
|
| D | convolution_transposed_4x4.h | 41 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
|
| D | convolution_transposed_3x3.h | 41 TuningType tuning_type, const GpuInfo& gpu_info,
|
| D | depthwise_conv_3x3.cc | 306 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
311 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | convolution_transposed_3x3.cc | 347 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
354 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | depthwise_conv_3x3.h | 42 TuningType tuning_type, const GpuInfo& gpu_info,
|
| D | depthwise_conv_3x3_stride_h2.h | 41 TuningType tuning_type, const GpuInfo& gpu_info,
|
| /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 | 78 "//tensorflow/lite/delegates/gpu/common/task:tuning_type",
174 name = "tuning_type",
175 hdrs = ["tuning_type.h"],
223 ":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.h | 135 void GetPossibleDispatches(TuningType tuning_type, const GpuInfo& gpu_info,
207 TuningType tuning_type, const GpuInfo& gpu_info,
|
| D | gpu_operation.cc | 438 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleDispatches() argument
442 GetPossibleKernelWorkGroups(tuning_type, gpu_info, kernel_info, in GetPossibleDispatches()
455 TuningType tuning_type, const GpuInfo& gpu_info, in GetPossibleKernelWorkGroups() argument
457 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_operation.cc | 191 absl::Status ClOperation::Tune(TuningType tuning_type, const GpuInfo& gpu_info, in Tune() argument
194 operation_->GetPossibleDispatches(tuning_type, gpu_info, kernel_.info_, in Tune()
|
| D | inference_context.cc | 286 TuningType tuning_type = TuningType::kExhaustive; in InitFromGpuModel() local
288 tuning_type = TuningType::kFast; in InitFromGpuModel()
295 tuning_type = TuningType::kFast; in InitFromGpuModel()
299 Tune(tuning_type, env->device().GetInfo(), env->profiling_queue())); in InitFromGpuModel()
682 absl::Status InferenceContext::Tune(TuningType tuning_type, in Tune() argument
687 node.cl_operation.Tune(tuning_type, gpu_info, profiling_queue)); in Tune()
|
| D | cl_operation.h | 115 absl::Status Tune(TuningType tuning_type, const GpuInfo& gpu_info,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | compute_task.h | 74 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
|
| D | compute_task.cc | 340 absl::Status ComputeTask::Tune(TuningType tuning_type, MetalDevice* device) { in Tune() argument
345 operation_->GetPossibleDispatches(tuning_type, device->GetInfo(), kernel_info, in Tune()
|
| D | BUILD | 121 "//tensorflow/lite/delegates/gpu/common/task:tuning_type",
168 "//tensorflow/lite/delegates/gpu/common/task:tuning_type",
|
| D | inference_context.h | 165 absl::Status Tune(TuningType tuning_type, MetalDevice* device);
|