Home
last modified time | relevance | path

Searched refs:add_2D_tensor_argument (Results 1 – 25 of 25) sorted by relevance

/external/ComputeLibrary/src/core/CL/kernels/
DCLL2NormalizeLayerKernel.cpp171 add_2D_tensor_argument(idx, _input, in_slice); in run()
172 add_2D_tensor_argument(idx, _sum, sum_slice); in run()
173 add_2D_tensor_argument(idx, _output, in_slice); in run()
187 add_2D_tensor_argument(idx, _input, in_slice); in run()
188 add_2D_tensor_argument(idx, _sum, sum_slice); in run()
189 add_2D_tensor_argument(idx, _output, in_slice); in run()
DCLBitwiseKernel.cpp107 add_2D_tensor_argument(idx, _input1, slice); in run()
110 add_2D_tensor_argument(idx, _input2, slice); in run()
112 add_2D_tensor_argument(idx, _output, slice); in run()
DCLArgMinMaxLayerKernel.cpp181 add_2D_tensor_argument(idx, _input, in_slice); in run()
184 add_2D_tensor_argument(idx, _prev_output, in_slice); in run()
186 add_2D_tensor_argument(idx, _output, out_slice); in run()
203 add_2D_tensor_argument(idx, _input, in_slice); in run()
204 add_2D_tensor_argument(idx, _output, out_slice); in run()
DCLBoundingBoxTransformKernel.cpp171 add_2D_tensor_argument(idx, _pred_boxes, slice); in run()
172 add_2D_tensor_argument(idx, _deltas, slice); in run()
DCLQLSTMLayerNormalizationKernel.cpp166 add_2D_tensor_argument(idx, _input, slice); in run()
169 add_2D_tensor_argument(idx, _output, slice); in run()
DCLReductionOperationKernel.cpp258 add_2D_tensor_argument(idx, _input, in_slice); in run()
259 add_2D_tensor_argument(idx, _output, out_slice); in run()
DCLMeanStdDevNormalizationKernel.cpp131 add_2D_tensor_argument(idx, _input, slice); in run()
DCLROIPoolingLayerKernel.cpp150 add_2D_tensor_argument(idx, _rois, slice_rois); in run()
DCLROIAlignLayerKernel.cpp167 add_2D_tensor_argument(idx, _rois, slice_rois); in run()
DCLPriorBoxLayerKernel.cpp224 add_2D_tensor_argument(idx, _output, slice); in run()
/external/ComputeLibrary/src/core/CL/
DICLSimple2DKernel.cpp40 add_2D_tensor_argument(idx, _input, slice); in run()
41 add_2D_tensor_argument(idx, _output, slice); in run()
DICLKernel.h189 void add_2D_tensor_argument(unsigned int &idx, const ICLTensor *tensor, const Window &window) in add_2D_tensor_argument() function
204 add_2D_tensor_argument(idx, tensor, window); in add_2D_tensor_argument_if()
/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/
DClKernelRuntime.cpp78 add_2D_tensor_argument(idx, tensor, arg_slice); in add_tensor_argument()
83 add_2D_tensor_argument(idx, tensor, arg_slice); in add_tensor_argument()
100 add_2D_tensor_argument(idx, tensor, arg_slice); in add_tensor_argument()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClTransposeKernel.cpp116 add_2D_tensor_argument(idx, src, slice); in run_op()
117 add_2D_tensor_argument(idx, dst, slice); in run_op()
DClConvertFullyConnectedWeightsKernel.cpp118 add_2D_tensor_argument(idx, src, window); in run_op()
119 add_2D_tensor_argument(idx, dst, window); in run_op()
DClGemmMatrixMultiplyNativeKernel.cpp435 add_2D_tensor_argument(idx, src0, slice); in run_op()
436 add_2D_tensor_argument(idx, src1, slice_b); in run_op()
439 add_2D_tensor_argument(idx, src2, slice); in run_op()
441 add_2D_tensor_argument(idx, dst, slice); in run_op()
446 add_2D_tensor_argument(idx, post_op_arg, slice); in run_op()
DClGemmLowpMatrixMultiplyNativeKernel.cpp326 add_2D_tensor_argument(idx, src0, slice); in run_op()
327 add_2D_tensor_argument(idx, src1, slice_b); in run_op()
328 add_2D_tensor_argument(idx, dst, slice); in run_op()
DClGemmLowpMatrixMultiplyReshapedKernel.cpp290 add_2D_tensor_argument(idx, src0, slice); in run_op()
291 add_2D_tensor_argument(idx, src1, slice_b); in run_op()
292 add_2D_tensor_argument(idx, dst, slice); in run_op()
DClGemmMatrixMultiplyReshapedKernel.cpp379 add_2D_tensor_argument(idx, src0, slice); in run_op()
388 add_2D_tensor_argument(idx, src1, slice_b); in run_op()
395 add_2D_tensor_argument(idx, dst, slice); in run_op()
401 add_2D_tensor_argument(idx, post_op_arg, slice); in run_op()
DClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp394 add_2D_tensor_argument(idx, src0, slice); in run_op()
403 add_2D_tensor_argument(idx, src1, slice_b); in run_op()
410 add_2D_tensor_argument(idx, dst, slice); in run_op()
416 add_2D_tensor_argument(idx, post_op_arg, slice); in run_op()
DClGemmLowpReductionKernel.cpp149 add_2D_tensor_argument(idx, dst, slice_out); in run_op()
220 add_2D_tensor_argument(idx, dst, slice_out); in run_op()
DClScaleKernel.cpp226 add_2D_tensor_argument(idx, src, slice); in run_op()
227 add_2D_tensor_argument(idx, dst, slice); in run_op()
DClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp524 add_2D_tensor_argument(idx, src0, slice); in run_op()
525 add_2D_tensor_argument(idx, src1, slice_b); in run_op()
526 add_2D_tensor_argument(idx, dst, slice); in run_op()
DClWeightsReshapeKernel.cpp150 add_2D_tensor_argument(idx, dst, out_slice); in run_op()
DClIm2ColKernel.cpp425 add_2D_tensor_argument(idx, dst, slice_out); in run_op()