24.02.1
|
#include "arm_compute/AclEntrypoints.h"
#include "src/common/ITensorV2.h"
#include "src/common/TensorPack.h"
#include "src/common/utils/Macros.h"
Go to the source code of this file.
Functions | |
AclStatus | AclCreateTensorPack (AclTensorPack *external_pack, AclContext external_ctx) |
Creates a tensor pack. More... | |
AclStatus | AclPackTensor (AclTensorPack external_pack, AclTensor external_tensor, int32_t slot_id) |
Add a tensor to a tensor pack. More... | |
AclStatus | AclPackTensors (AclTensorPack external_pack, AclTensor *external_tensors, int32_t *slot_ids, size_t num_tensors) |
A list of tensors to a tensor pack. More... | |
AclStatus | AclDestroyTensorPack (AclTensorPack external_pack) |
Destroy a given tensor pack object. More... | |
AclStatus AclCreateTensorPack | ( | AclTensorPack * | pack, |
AclContext | ctx | ||
) |
Creates a tensor pack.
Tensor packs are used to create a collection of tensors that can be passed around for operator execution
[in,out] | pack | A valid non-zero tensor pack object if no failures occur |
[in] | ctx | Context to be used |
Returns:
Definition at line 51 of file AclTensorPack.cpp.
References AclOutOfMemory, AclSuccess, ARM_COMPUTE_LOG_ERROR_WITH_FUNCNAME_ACL, ARM_COMPUTE_RETURN_CENUM_ON_FAILURE, arm_compute::get_internal(), arm_compute::test::validation::pack, and arm_compute::detail::validate_internal_context().
Referenced by TensorPack::TensorPack().
AclStatus AclDestroyTensorPack | ( | AclTensorPack | pack | ) |
Destroy a given tensor pack object.
[in,out] | pack | A valid tensor pack object to destroy |
Returns:
Definition at line 96 of file AclTensorPack.cpp.
References AclSuccess, ARM_COMPUTE_RETURN_CENUM_ON_FAILURE, arm_compute::get_internal(), arm_compute::test::validation::pack, and arm_compute::detail::validate_internal_pack().
AclStatus AclPackTensor | ( | AclTensorPack | pack, |
AclTensor | tensor, | ||
int32_t | slot_id | ||
) |
Add a tensor to a tensor pack.
[in,out] | pack | Pack to append a tensor to |
[in] | tensor | Tensor to pack |
[in] | slot_id | Slot of the operator that the tensors corresponds to |
Returns:
Definition at line 71 of file AclTensorPack.cpp.
References AclSuccess, ARM_COMPUTE_RETURN_CENUM_ON_FAILURE, arm_compute::get_internal(), arm_compute::test::validation::pack, and arm_compute::detail::validate_internal_pack().
Referenced by TensorPack::add().
AclStatus AclPackTensors | ( | AclTensorPack | pack, |
AclTensor * | tensors, | ||
int32_t * | slot_ids, | ||
size_t | num_tensors | ||
) |
A list of tensors to a tensor pack.
[in,out] | pack | Pack to append the tensors to |
[in] | tensors | Tensors to append to the pack |
[in] | slot_ids | Slot IDs of each tensors to the operators |
[in] | num_tensors | Number of tensors that are passed |
Returns:
Definition at line 82 of file AclTensorPack.cpp.
References AclSuccess, ARM_COMPUTE_RETURN_CENUM_ON_FAILURE, arm_compute::get_internal(), arm_compute::test::validation::pack, and arm_compute::detail::validate_internal_pack().
Referenced by TensorPack::add().