Searched refs:grid_dimension_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_operation.cc | 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 | 820 op->grid_dimension_ = fb_op->grid_dimension(); in Decode()
887 op_builder.add_grid_dimension(op.grid_dimension_); in Encode()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | gpu_operation.cc | 104 grid_dimension_(operation.grid_dimension_), in GPUOperation()
126 std::swap(grid_dimension_, operation.grid_dimension_); in operator =()
|
| D | gpu_operation.h | 171 int grid_dimension_ = 3; // can be 1, 2 or 3 variable
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | compute_task.cc | 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 | conv_metal.cc | 1066 desc.grid_dimension_ = 1; in CreateConvolutionMetal()
1068 desc.grid_dimension_ = 2; in CreateConvolutionMetal()
1070 desc.grid_dimension_ = 3; in CreateConvolutionMetal()
1180 desc.grid_dimension_ = 1; in CreateConvolutionMetalWino4x4To6x6()
1182 desc.grid_dimension_ = 2; in CreateConvolutionMetalWino4x4To6x6()
1184 desc.grid_dimension_ = 3; in CreateConvolutionMetalWino4x4To6x6()
|
| D | conv_powervr.cc | 243 grid_dimension_ = 1; in GenerateCode()
245 grid_dimension_ = 2; in GenerateCode()
|