lhutton1 commented on a change in pull request #6109:
URL: https://github.com/apache/incubator-tvm/pull/6109#discussion_r459624243
##########
File path: src/runtime/contrib/arm_compute_lib/acl_utils.cc
##########
@@ -108,6 +111,30 @@ arm_compute::PadStrideInfo ToACLPadStride(const
std::vector<std::string>& pad,
arm_compute::DimensionRoundingType::FLOOR);
}
+arm_compute::DataType MakeDataType(const DLDataType& data_type) {
+ if (data_type.code == DLDataTypeCode::kDLFloat && data_type.bits == 32) {
+ return arm_compute::DataType::F32;
+ } else if (data_type.code == DLDataTypeCode::kDLUInt && data_type.bits == 8)
{
+ return arm_compute::DataType::QASYMM8;
+ } else if (data_type.code == DLDataTypeCode::kDLInt && data_type.bits == 32)
{
+ return arm_compute::DataType::S32;
+ } else {
+ LOG(FATAL) << "Datatype " << data_type << " unsupported by ACL runtime";
+ return arm_compute::DataType::UNKNOWN;
+ }
+}
+
+template <typename T>
+std::vector<T> GetVectorFromDLTensor(const DLTensor* tensor) {
Review comment:
Its not so much about flattening the tensor but rather converting it to
the type that ACL expects. `arm_compute::QuantizationInfo(std::vector<float>
scale, std::vector<int32_t> offset)`
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
