Searched refs:add_3d_tensor_nhw_argument (Results 1 – 6 of 6) sorted by relevance
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClGemmLowpMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 446 add_3d_tensor_nhw_argument(idx, src0); in run_op()
447 add_3d_tensor_nhw_argument(idx, src1); in run_op()
452 add_3d_tensor_nhw_argument(idx, src2); in run_op()
455 add_3d_tensor_nhw_argument(idx, dst); in run_op()
466 add_3d_tensor_nhw_argument(idx, vector_sum_col); in run_op()
470 add_3d_tensor_nhw_argument(idx, vector_sum_row); in run_op()
|
| D | ClGemmMatrixMultiplyReshapedOnlyRhsMMULKernel.cpp | 337 add_3d_tensor_nhw_argument(idx, src0); in run_op()
342 add_3d_tensor_nhw_argument(idx, src1); in run_op()
347 add_3d_tensor_nhw_argument(idx, src2); in run_op()
350 add_3d_tensor_nhw_argument(idx, dst); in run_op()
|
| D | ClGemmReshapeRhsMatrixKernel.cpp | 169 add_3d_tensor_nhw_argument(idx, src); in run_op()
170 add_3d_tensor_nhw_argument(idx, dst); in run_op()
|
| D | ClGemmReshapeLhsMatrixKernel.cpp | 192 add_3d_tensor_nhw_argument(idx, src); in run_op()
193 add_3d_tensor_nhw_argument(idx, dst); in run_op()
|
| /external/ComputeLibrary/src/core/CL/ |
| D | ICLKernel.cpp | 119 void ICLKernel::add_3d_tensor_nhw_argument(unsigned int &idx, const ICLTensor *tensor) in add_3d_tensor_nhw_argument() function in arm_compute::ICLKernel
|
| D | ICLKernel.h | 243 void add_3d_tensor_nhw_argument(unsigned int &idx, const ICLTensor *tensor);
|