Home
last modified time | relevance | path

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

12

/external/ComputeLibrary/src/core/CL/kernels/
DCLFuseBatchNormalizationKernel.cpp206 add_1D_tensor_argument(idx, _input_bias, slice_1d); in run()
208 add_1D_tensor_argument(idx, _bn_mean, slice_1d); in run()
209 add_1D_tensor_argument(idx, _bn_var, slice_1d); in run()
216 add_1D_tensor_argument(idx, _fused_bias, slice_1d); in run()
220 add_1D_tensor_argument(idx, _bn_beta, slice_1d); in run()
224 add_1D_tensor_argument(idx, _bn_gamma, slice_1d); in run()
DCLBatchNormalizationLayerKernel.cpp234 add_1D_tensor_argument(idx, _mean, vector_slice); in run()
235 add_1D_tensor_argument(idx, _var, vector_slice); in run()
238 add_1D_tensor_argument(idx, _beta, vector_slice); in run()
242 add_1D_tensor_argument(idx, _gamma, vector_slice); in run()
DCLGenerateProposalsLayerKernel.cpp145 add_1D_tensor_argument(idx, _anchors, collapsed); in run()
146 add_1D_tensor_argument(idx, _all_anchors, collapsed); in run()
DCLQLSTMLayerNormalizationKernel.cpp167 add_1D_tensor_argument(idx, _weight, weight_slice); in run()
168 add_1D_tensor_argument(idx, _bias, weight_slice); in run()
DCLNormalizePlanarYUVLayerKernel.cpp188 add_1D_tensor_argument(idx, _mean, slice_in); in run()
189 add_1D_tensor_argument(idx, _std, slice_in); in run()
DCLReductionOperationKernel.cpp224 add_1D_tensor_argument(idx, _input, in_slice); in run()
225 add_1D_tensor_argument(idx, _output, out_slice); in run()
DCLDepthwiseConvolutionLayerNativeKernel.cpp408 add_1D_tensor_argument(idx, _output_multipliers, slice); in run()
409 add_1D_tensor_argument(idx, _output_shifts, slice); in run()
413 add_1D_tensor_argument(idx, _biases, slice); in run()
DCLReverseKernel.cpp136 add_1D_tensor_argument(idx, _axis, axis_slice); in run()
DCLRangeKernel.cpp135 add_1D_tensor_argument(idx, _output, window); in run()
DCLGatherKernel.cpp129 add_1D_tensor_argument(idx, _indices, window_collapsed); in run()
DCLFFTDigitReverseKernel.cpp136 add_1D_tensor_argument(idx, _idx, slice); in run()
DCLSelectKernel.cpp149 add_1D_tensor_argument(idx, _c, vector_slice); in run()
DCLBoundingBoxTransformKernel.cpp170 add_1D_tensor_argument(idx, _boxes, slice); in run()
DCLBatchToSpaceLayerKernel.cpp193 add_1D_tensor_argument(idx, _block_shape, vector_slice); in run()
DCLDeconvolutionReshapeOutputKernel.cpp199 add_1D_tensor_argument(idx, _bias, collapsed); in run()
/external/ComputeLibrary/src/core/CL/
DICLKernel.h165 void add_1D_tensor_argument(unsigned int &idx, const ICLTensor *tensor, const Window &window) in add_1D_tensor_argument() function
180 add_1D_tensor_argument(idx, tensor, window); in add_1D_tensor_argument_if()
/external/ComputeLibrary/src/gpu/cl/kernels/
DClGemmLowpQuantizeDownInt32ScaleByFloatKernel.cpp151 add_1D_tensor_argument(idx1, bias, biases_slice); in run_op()
DClGemmLowpQuantizeDownInt32ScaleByFixedPointKernel.cpp150 add_1D_tensor_argument(idx1, bias, biases_slice); in run_op()
DClWeightsReshapeKernel.cpp153 add_1D_tensor_argument(idx, biases, biases_slice); in run_op()
DClGemmLowpQuantizeDownInt32ScaleKernel.cpp148 add_1D_tensor_argument(idx1, bias, biases_slice); in run_op()
DClDirectConv2dKernel.cpp472 add_1D_tensor_argument(idx, biases, slice); in run_op()
485 add_1D_tensor_argument(idx1, biases, slice_biases); in run_op()
DClTransposedConvolutionKernel.cpp263 add_1D_tensor_argument(idx, biases, slice); in run_op()
DClIndirectConv2dKernel.cpp299 add_1D_tensor_argument(idx, biases, slice); in run_op()
DClDirectConv3dKernel.cpp246 add_1D_tensor_argument(idx, biases, slice); in run_op()
/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/
DClKernelRuntime.cpp72 add_1D_tensor_argument(idx, tensor, arg_slice); in add_tensor_argument()

12