diff options
Diffstat (limited to 'tests/nnfw_api/src/CircleGen.h')
-rw-r--r-- | tests/nnfw_api/src/CircleGen.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tests/nnfw_api/src/CircleGen.h b/tests/nnfw_api/src/CircleGen.h index 666218379..2b88af7b7 100644 --- a/tests/nnfw_api/src/CircleGen.h +++ b/tests/nnfw_api/src/CircleGen.h @@ -128,6 +128,8 @@ public: uint32_t addBuffer(const uint8_t *buf, size_t size); uint32_t addTensor(const TensorParams ¶ms); uint32_t addTensor(const TensorParams ¶ms, float scale, int64_t zero_point); + uint32_t addTensor(const TensorParams ¶ms, std::vector<float> &scale, + std::vector<int64_t> &zero_point); uint32_t addTensor(const TensorParams ¶ms, const SparsityParams &sp); void setInputsAndOutputs(const std::vector<int> &inputs, const std::vector<int> &outputs); uint32_t nextSubgraph(); @@ -172,11 +174,13 @@ public: uint32_t addOperatorLeakyRelu(const OperatorParams ¶ms, float alpha); uint32_t addOperatorLess(const OperatorParams ¶ms); uint32_t addOperatorLogSoftmax(const OperatorParams ¶ms); + uint32_t addOperatorMul(const OperatorParams ¶ms, circle::ActivationFunctionType actfn); uint32_t addOperatorMean(const OperatorParams ¶ms, bool keep_dims); uint32_t addOperatorNeg(const OperatorParams ¶ms); uint32_t addOperatorOneHot(const OperatorParams ¶ms, int32_t axis); uint32_t addOperatorPad(const OperatorParams ¶ms); uint32_t addOperatorPadV2(const OperatorParams ¶ms); + uint32_t addOperatorQuantize(const OperatorParams ¶ms); uint32_t addOperatorRank(const OperatorParams ¶ms); uint32_t addOperatorReduce(const OperatorParams ¶ms, circle::BuiltinOperator reduce_op, bool keep_dims); @@ -193,12 +197,15 @@ public: circle::TensorType type = circle::TensorType::TensorType_INT32); uint32_t addOperatorSelect(const OperatorParams ¶ms); uint32_t addOperatorSelectV2(const OperatorParams ¶ms); + uint32_t addOperatorSlice(const OperatorParams ¶ms); + uint32_t addOperatorSoftmax(const OperatorParams ¶ms, float beta); uint32_t addOperatorSplit(const OperatorParams ¶ms, int32_t num_split); uint32_t addOperatorSqrt(const OperatorParams ¶ms); uint32_t addOperatorSquare(const OperatorParams ¶ms); uint32_t addOperatorStridedSlice(const OperatorParams ¶ms, int32_t begin_mask = 0, int32_t end_mask = 0, int32_t ellipsis_mask = 0, int32_t new_axis_mask = 0, int32_t shrink_axis_mask = 0); + uint32_t addOperatorSub(const OperatorParams ¶ms, circle::ActivationFunctionType actfn); uint32_t addOperatorTile(const OperatorParams ¶ms); uint32_t addOperatorTranspose(const OperatorParams ¶ms); uint32_t addOperatorWhile(const OperatorParams ¶ms, uint32_t cond_subg, uint32_t body_subg); @@ -215,6 +222,9 @@ private: flatbuffers::Offset<circle::Tensor> buildTensor(const TensorParams ¶ms); flatbuffers::Offset<circle::Tensor> buildTensor(const TensorParams ¶ms, float scale, int64_t zero_point); + flatbuffers::Offset<circle::Tensor> buildTensor(const TensorParams ¶ms, + std::vector<float> &scales, + std::vector<int64_t> &zero_points); flatbuffers::Offset<circle::SparsityParameters> buildSparsityParameters(const SparsityParams &sp); flatbuffers::Offset<circle::Tensor> buildTensor(const TensorParams ¶ms, const SparsityParams &sp); |