Home
last modified time | relevance | path

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

12

/external/ComputeLibrary/src/gpu/cl/kernels/
DClWidthConcatenate4TensorsKernel.cpp178 add_4D_tensor_argument(idx, src0, slice); in run_op()
179 add_4D_tensor_argument(idx, src1, slice); in run_op()
180 add_4D_tensor_argument(idx, src2, slice); in run_op()
181 add_4D_tensor_argument(idx, src3, slice); in run_op()
182 add_4D_tensor_argument(idx, dst, slice); in run_op()
DClWidthConcatenate2TensorsKernel.cpp155 add_4D_tensor_argument(idx, src0, slice); in run_op()
156 add_4D_tensor_argument(idx, src1, slice); in run_op()
157 add_4D_tensor_argument(idx, dst, slice); in run_op()
DClHeightConcatenateKernel.cpp131 add_4D_tensor_argument(idx, src, window); in run_op()
132 add_4D_tensor_argument(idx, dst, window); in run_op()
DClWidthConcatenateKernel.cpp126 add_4D_tensor_argument(idx, src, window); in run_op()
127 add_4D_tensor_argument(idx, dst, window); in run_op()
DClPermuteKernel.cpp144 add_4D_tensor_argument(idx, src, slice_in); in run_op()
145 add_4D_tensor_argument(idx, dst, slice_out); in run_op()
DClDirectConv3dKernel.cpp241 add_4D_tensor_argument(idx, src, slice); in run_op()
242 add_4D_tensor_argument(idx, dst, slice); in run_op()
243 add_4D_tensor_argument(idx, weights, slice); in run_op()
DClPool2dKernel.cpp346 add_4D_tensor_argument(idx, src, in_slice); in run_op()
347 add_4D_tensor_argument(idx, dst, slice); in run_op()
350 add_4D_tensor_argument(idx, indices, slice); in run_op()
DClWinogradInputTransformKernel.cpp248 add_4D_tensor_argument(idx, src, slice); in run_op()
249 add_4D_tensor_argument(idx, dst, slice); in run_op()
DClWinogradOutputTransformKernel.cpp305 add_4D_tensor_argument(idx, src, slice); in run_op()
306 add_4D_tensor_argument(idx, dst, slice_out); in run_op()
DClWinogradFilterTransformKernel.cpp162 add_4D_tensor_argument(idx, src, window); in run_op()
DClCol2ImKernel.cpp168 add_4D_tensor_argument(idx, dst, slice_out); in run_op()
/external/ComputeLibrary/src/core/CL/kernels/
DCLReverseKernel.cpp135 add_4D_tensor_argument(idx, _input, slice); in run()
137 add_4D_tensor_argument(idx, _output, slice); in run()
DCLInstanceNormalizationLayerKernel.cpp147 add_4D_tensor_argument(idx, _input, collapsed_window); in run()
225 add_4D_tensor_argument(idx, _input, collapsed_window); in run()
230 add_4D_tensor_argument(idx, _output, collapsed_window); in run()
DCLStackLayerKernel.cpp136 add_4D_tensor_argument(idx, _input, slice_in); in run()
137 add_4D_tensor_argument(idx, _output, slice_out); in run()
DCLGatherKernel.cpp128 add_4D_tensor_argument(idx, _input, window_collapsed); in run()
130 add_4D_tensor_argument(idx, _output, window_collapsed); in run()
DCLTileKernel.cpp146 add_4D_tensor_argument(idx, _input, slice); in run()
147 add_4D_tensor_argument(idx, _output, slice); in run()
DCLChannelShuffleLayerKernel.cpp190 add_4D_tensor_argument(idx, _input, window); in run()
191 add_4D_tensor_argument(idx, _output, window); in run()
DCLStridedSliceKernel.cpp181 add_4D_tensor_argument(idx, src, slice); in run_op()
182 add_4D_tensor_argument(idx, dst, slice); in run_op()
DCLArgMinMaxLayerKernel.cpp239 add_4D_tensor_argument(idx, _input, in_slice); in run()
240 add_4D_tensor_argument(idx, _output, out_slice); in run()
DCLReductionOperationKernel.cpp294 add_4D_tensor_argument(idx, _input, in_slice); in run()
295 add_4D_tensor_argument(idx, _output, out_slice); in run()
DCLDepthToSpaceLayerKernel.cpp135 add_4D_tensor_argument(idx, _output, slice_out); in run()
DCLSpaceToDepthLayerKernel.cpp132 add_4D_tensor_argument(idx, _input, slice_in); in run()
DCLBatchToSpaceLayerKernel.cpp195 add_4D_tensor_argument(idx, _output, slice_out); in run()
/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/
DClKernelRuntime.cpp123 add_4D_tensor_argument(idx, tensor, arg_slice); in add_tensor_argument()
/external/ComputeLibrary/src/core/CL/
DICLKernel.h223 void add_4D_tensor_argument(unsigned int &idx, const ICLTensor *tensor, const Window &window) in add_4D_tensor_argument() function

12