diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-11-02 17:16:25 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-11-10 16:35:17 +0000 |
commit | aee62afba99a74f772b97356fd4c18f3fdf37073 (patch) | |
tree | 47c7f0bc619ae44332f58c786b52460b40bc1a97 /reference_model/src/generate | |
parent | bfc53031803338d9f0866f88f1d2deffd4928bcc (diff) | |
download | reference_model-aee62afba99a74f772b97356fd4c18f3fdf37073.tar.gz |
Main Compliance testing for FULLY_CONNECTED
Updated shapes to meet MIN_DOT_PRODUCTS.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I82297917c009b3120306f8a9bb965209d109ef8d
Diffstat (limited to 'reference_model/src/generate')
-rw-r--r-- | reference_model/src/generate/generate_dot_product.cc | 98 | ||||
-rw-r--r-- | reference_model/src/generate/generate_utils.cc | 1 |
2 files changed, 99 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_dot_product.cc b/reference_model/src/generate/generate_dot_product.cc index c1934dd..0930956 100644 --- a/reference_model/src/generate/generate_dot_product.cc +++ b/reference_model/src/generate/generate_dot_product.cc @@ -231,6 +231,102 @@ bool generateReduceSum(const TosaReference::GenerateConfig& cfg, } return true; } +//---------------------------------------------------------------------------// +// Fully Connected // +//---------------------------------------------------------------------------// + +bool generateFullyConnectedInput(const TosaReference::GenerateConfig& cfg, + TosaReference::IDotProductGenerator& generator, + void* data, + size_t size) +{ + if (cfg.shape.size() != 2) + { + WARNING("[Generator][DP][FullyConnected][Input] Tensor shape expected 2 dimensions."); + return false; + } + + float* input = reinterpret_cast<float*>(data); + const int64_t T = TosaReference::numElementsFromShape(cfg.shape); + const uint32_t IC = cfg.shape[1]; + + for (int64_t t = 0; t < T; ++t) + { + uint32_t k = t % IC; + + input[t] = generator(k); + } + return true; +} + +bool generateFullyConnectedWeight(const TosaReference::GenerateConfig& cfg, + TosaReference::IDotProductGenerator& generator, + void* data, + size_t size) +{ + if (cfg.shape.size() != 2) + { + WARNING("[Generator][DP][FullyConnected][Weight] Tensor shape expected 2 dimensions."); + return false; + } + + float* weight = reinterpret_cast<float*>(data); + const int64_t T = TosaReference::numElementsFromShape(cfg.shape); + const uint32_t IC = cfg.shape[1]; + + for (int64_t t = 0; t < T; ++t) + { + uint32_t k = t % IC; + + weight[t] = generator(k); + } + return true; +} + +bool generateFullyConnectedBias(const TosaReference::GenerateConfig& cfg, + TosaReference::IDotProductGenerator& generator, + void* data, + size_t size) +{ + if (cfg.shape.size() != 1) + { + WARNING("[Generator][DP][FullyConnected][Bias] Tensor shape expected 1 dimension."); + return false; + } + + float* bias = reinterpret_cast<float*>(data); + const uint32_t T = cfg.shape[0]; + + for (uint32_t t = 0; t < T; ++t) + { + bias[t] = generator(2); + } + return true; +} + +bool generateFullyConnected(const TosaReference::GenerateConfig& cfg, + TosaReference::IDotProductGenerator& generator, + void* data, + size_t size) +{ + if (cfg.dataType != DType::DType_FP32) + { + WARNING("[Generator][DP][FullyConnected] Only supports FP32."); + return false; + } + switch (cfg.inputPos) + { + case 0: + return generateFullyConnectedInput(cfg, generator, data, size); + case 1: + return generateFullyConnectedWeight(cfg, generator, data, size); + case 2: + return generateFullyConnectedBias(cfg, generator, data, size); + default: + WARNING("[Generator][DP][FullyConnected] Invalid input tensor slot position to operator."); + return false; + } +} } // namespace namespace TosaReference @@ -254,6 +350,8 @@ bool generateDotProduct(const GenerateConfig& cfg, void* data, size_t size) return generateConv2D(cfg, *generator, data, size); case tosa::Op_REDUCE_SUM: return generateReduceSum(cfg, *generator, data, size); + case tosa::Op_FULLY_CONNECTED: + return generateFullyConnected(cfg, *generator, data, size); default: WARNING("[Generator][DP] Unsupported operator."); return false; diff --git a/reference_model/src/generate/generate_utils.cc b/reference_model/src/generate/generate_utils.cc index e410436..c889f7b 100644 --- a/reference_model/src/generate/generate_utils.cc +++ b/reference_model/src/generate/generate_utils.cc @@ -41,6 +41,7 @@ NLOHMANN_JSON_SERIALIZE_ENUM(Op, { Op::Op_ADD, "ADD" }, { Op::Op_ARGMAX, "ARGMAX" }, { Op::Op_CONV2D, "CONV2D" }, + { Op::Op_FULLY_CONNECTED, "FULLY_CONNECTED" }, { Op::Op_MATMUL, "MATMUL" }, { Op::Op_MAXIMUM, "MAXIMUM" }, { Op::Op_MAX_POOL2D, "MAX_POOL2D" }, |