Searched refs:grid_size_ (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.cc | 106 grid_size_(operation.grid_size_), in GPUOperation()
128 std::swap(grid_size_, operation.grid_size_); in operator =()
212 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
241 return grid_size_; in GetGridSize()
|
| D | gpu_operation.h | 173 int3 grid_size_ = int3(0, 0, 0); variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_operation.cc | 176 operation_->grid_size_ = operation_->GetGridSize(); in UpdateParams()
178 operation_->grid_dimension_, operation_->grid_size_, in UpdateParams()
220 operation_->grid_dimension_, operation_->grid_size_, in Tune()
227 operation_->grid_dimension_, operation_->grid_size_, in Tune()
237 operation_->grid_dimension_, operation_->grid_size_, in Tune()
|
| D | serialization.cc | 824 op->grid_size_.x = fb_op->grid_size()->x(); in Decode()
825 op->grid_size_.y = fb_op->grid_size()->y(); in Decode()
826 op->grid_size_.z = fb_op->grid_size()->z(); in Decode()
860 auto grid_size_fb = Encode(op.grid_size_, builder); in Encode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | compute_task.cc | 210 operation_->grid_size_ = operation_->GetGridSize(); in UpdateParams()
212 operation_->grid_dimension_, operation_->grid_size_, in UpdateParams()
255 operation_->grid_dimension_, operation_->grid_size_, in Tune()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv_3x3_stride_h2.cc | 203 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | winograd.cc | 476 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
690 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | depthwise_conv_3x3.cc | 297 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | convolution_transposed_3x3.cc | 360 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | reduce.cc | 470 GetPossibleWorkGroups(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | conv_buffer_1x1.cc | 317 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | convolution_transposed.cc | 574 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|
| D | conv_powervr.cc | 333 GetPossibleWorkGroupsConv(tuning_type, gpu_info, kernel_info, grid_size_, in GetPossibleKernelWorkGroups()
|