Lines Matching refs:get_tensor
153 auto dst = tensors.get_tensor(TensorType::ACL_DST_0); in run()
154 auto workspace = tensors.get_tensor(TensorType::ACL_INT_3); in run()
155 auto packed_weights = tensors.get_tensor(TensorType::ACL_INT_4); in run()
162 auto src_perm = tensors.get_tensor(TensorType::ACL_INT_0); in run()
171 auto src_perm = tensors.get_tensor(TensorType::ACL_INT_0); in run()
172 auto weights_perm = tensors.get_tensor(TensorType::ACL_INT_1); in run()
173 auto dst_perm = tensors.get_tensor(TensorType::ACL_INT_2); in run()
186 auto src = tensors.get_tensor(TensorType::ACL_SRC_0); in run()
187 auto weights = tensors.get_tensor(TensorType::ACL_SRC_1); in run()
188 auto dst = tensors.get_tensor(TensorType::ACL_DST); in run()
204 auto dst_perm = tensors.get_tensor(TensorType::ACL_INT_2); in run()
228 auto packed_weights = tensors.get_tensor(TensorType::ACL_INT_4); in prepare()
245 auto packed_weights = tensors.get_tensor(TensorType::ACL_INT_4); in prepare()
250 auto permuted_weights = tensors.get_tensor(TensorType::ACL_INT_1); in prepare()
376 auto dst = tensors.get_tensor(TensorType::ACL_DST_0); in run()
381 auto src_perm = tensors.get_tensor(TensorType::ACL_INT_0); in run()
382 auto weights_perm = tensors.get_tensor(TensorType::ACL_INT_1); in run()
383 auto dst_perm = tensors.get_tensor(TensorType::ACL_INT_2); in run()
410 auto dst_perm = tensors.get_tensor(TensorType::ACL_INT_2); in run()
430 auto weights_perm = tensors.get_tensor(TensorType::ACL_INT_1); in prepare()