Searched refs:add_2D_tensor_argument_if (Results 1 – 8 of 8) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClGemmLowpOffsetContributionKernel.cpp | 205 … add_2D_tensor_argument_if((vector_sum_col != nullptr), idx, vector_sum_col, win_vector_sum_col); in run_op()
206 … add_2D_tensor_argument_if((vector_sum_row != nullptr), idx, vector_sum_row, win_vector_sum_row); in run_op()
|
| D | ClGemmLowpOffsetContributionOutputStageKernel.cpp | 254 … add_2D_tensor_argument_if((vector_sum_col != nullptr), idx, vector_sum_col, win_vector_sum_col); in run_op()
255 … add_2D_tensor_argument_if((vector_sum_row != nullptr), idx, vector_sum_row, win_vector_sum_row); in run_op()
|
| D | ClGemmLowpMatrixMultiplyReshapedOnlyRhsKernel.cpp | 544 … add_2D_tensor_argument_if((vector_sum_col != nullptr), idx, vector_sum_col, win_vector_sum_col); in run_op()
545 … add_2D_tensor_argument_if((vector_sum_row != nullptr), idx, vector_sum_row, win_vector_sum_row); in run_op()
|
| D | ClGemmMatrixMultiplyReshapedKernel.cpp | 392 add_2D_tensor_argument_if(_add_bias, idx, src2, slice); in run_op()
|
| D | ClGemmMatrixMultiplyReshapedOnlyRhsKernel.cpp | 407 add_2D_tensor_argument_if(_add_bias, idx, src2, slice); in run_op()
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLMeanStdDevNormalizationKernel.cpp | 132 add_2D_tensor_argument_if((!_run_in_place), idx, _output, slice); in run()
|
| D | CLSpaceToBatchLayerKernel.cpp | 213 add_2D_tensor_argument_if(cond, idx, _paddings, padding_slice); in run()
|
| /external/ComputeLibrary/src/core/CL/ |
| D | ICLKernel.h | 200 …void add_2D_tensor_argument_if(bool cond, unsigned int &idx, const ICLTensor *tensor, const Window… in add_2D_tensor_argument_if() function
|