Home
last modified time | relevance | path

Searched refs:work_group_launch_order_ (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed_4x4.cc54 work_group_launch_order_ = int3(2, 0, 1); in ConvolutionTransposed4x4()
156 launch_remap[work_group_launch_order_.x] = 0; in GenerateConvolutionTransposedCode()
157 launch_remap[work_group_launch_order_.y] = 1; in GenerateConvolutionTransposedCode()
158 launch_remap[work_group_launch_order_.z] = 2; in GenerateConvolutionTransposedCode()
159 if (work_group_launch_order_[0] == 0) { in GenerateConvolutionTransposedCode()
165 if (work_group_launch_order_[1] == 1) { in GenerateConvolutionTransposedCode()
171 if (work_group_launch_order_[2] == 2) { in GenerateConvolutionTransposedCode()
Dconv_powervr.cc511 c += GenerateBlockCoords(conv_params.block_size, work_group_launch_order_, in GenerateConv()
1074 work_group_launch_order_ = int3(2, 0, 1); in GuessBestParams()
1079 work_group_launch_order_ = int3(1, 0, 2); in GuessBestParams()
1119 work_group_launch_order_ = int3(2, 0, 1); in GuessBestParams()
1124 work_group_launch_order_ = int3(1, 0, 2); in GuessBestParams()
1167 work_group_launch_order_ = int3(2, 0, 1); in GuessBestParams()
1171 work_group_launch_order_ = int3(2, 0, 1); in GuessBestParams()
1230 work_group_launch_order_ = int3(0, 1, 2); in GuessBestParams()
1245 work_group_launch_order_ = int3(0, 1, 2); in GuessBestParams()
1257 work_group_launch_order_ = int3(0, 1, 2); in GuessBestParams()
[all …]
Dconvolution_transposed_3x3.cc31 work_group_launch_order_ = int3(2, 0, 1); in ConvolutionTransposed3x3()
48 work_group_launch_order_); in ConvolutionTransposed3x3()
Dconv_metal.cc1064 desc.work_group_launch_order_ = params.work_group_launch_order; in CreateConvolutionMetal()
1178 desc.work_group_launch_order_ = params.work_group_launch_order; in CreateConvolutionMetalWino4x4To6x6()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_operation.cc179 operation_->work_group_size_, operation_->work_group_launch_order_); in UpdateParams()
221 operation_->work_group_size_, operation_->work_group_launch_order_); in Tune()
228 possible_work_groups[i], operation_->work_group_launch_order_); in Tune()
238 operation_->work_group_size_, operation_->work_group_launch_order_); in Tune()
Dserialization.cc821 op->work_group_launch_order_.x = fb_op->work_group_launch_order()->x(); in Decode()
822 op->work_group_launch_order_.y = fb_op->work_group_launch_order()->y(); in Decode()
823 op->work_group_launch_order_.z = fb_op->work_group_launch_order()->z(); in Decode()
859 Encode(op.work_group_launch_order_, builder); in Encode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dgpu_operation.cc105 work_group_launch_order_(operation.work_group_launch_order_), in GPUOperation()
127 std::swap(work_group_launch_order_, operation.work_group_launch_order_); in operator =()
Dgpu_operation.h172 int3 work_group_launch_order_ = int3(0, 1, 2); variable
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dcompute_task.cc213 operation_->work_group_size_, operation_->work_group_launch_order_); in UpdateParams()
256 operation_->work_group_size_, operation_->work_group_launch_order_); in Tune()