diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2022-09-07 19:04:21 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2022-09-07 19:04:21 +0900 |
commit | c690d52bdd137ed6a17353aa7af35e8141ece77b (patch) | |
tree | dbb7dd99133132dfbffcb8c9e9af4f1ffc2f4808 /compiler/circle-tensordump/src/Reader.h | |
parent | 3ad689f0803519e343c36d5700646e86059df961 (diff) | |
download | nnfw-tizen_7.0_hotfix.tar.gz nnfw-tizen_7.0_hotfix.tar.bz2 nnfw-tizen_7.0_hotfix.zip |
Imported Upstream version 1.21.0upstream/1.21.0tizen_7.0_m2_releaseaccepted/tizen/unified/20220912.170817accepted/tizen/unified/20220912.164738accepted/tizen/7.0/unified/hotfix/20221116.105341accepted/tizen/7.0/unified/20221110.060236tizen_7.0_hotfixtizen_7.0accepted/tizen_7.0_unified_hotfixaccepted/tizen_7.0_unified
Diffstat (limited to 'compiler/circle-tensordump/src/Reader.h')
-rw-r--r-- | compiler/circle-tensordump/src/Reader.h | 85 |
1 files changed, 0 insertions, 85 deletions
diff --git a/compiler/circle-tensordump/src/Reader.h b/compiler/circle-tensordump/src/Reader.h deleted file mode 100644 index c868bc277..000000000 --- a/compiler/circle-tensordump/src/Reader.h +++ /dev/null @@ -1,85 +0,0 @@ -/* - * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved - * - * Licensed under the Apache License, Version 2.0 (the "License"); - * you may not use this file except in compliance with the License. - * You may obtain a copy of the License at - * - * http://www.apache.org/licenses/LICENSE-2.0 - * - * Unless required by applicable law or agreed to in writing, software - * distributed under the License is distributed on an "AS IS" BASIS, - * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. - * See the License for the specific language governing permissions and - * limitations under the License. - */ - -#ifndef __CIRCLE_TENSORDUMP_READER_H__ -#define __CIRCLE_TENSORDUMP_READER_H__ - -#include <mio/circle/schema_generated.h> - -#include <map> -#include <string> -#include <vector> - -namespace circletensordump -{ - -template <typename T> std::vector<T> as_index_vector(const flatbuffers::Vector<T> *flat_array) -{ - std::vector<T> ret(flat_array->Length()); - for (uint32_t i = 0; i < flat_array->Length(); i++) - { - ret[i] = flat_array->Get(i); - } - return ret; -} - -/** - * @brief Loads Circle file and provides helpers to access attributes - */ -class Reader -{ -private: - using CircleSubGraphs_t = flatbuffers::Vector<flatbuffers::Offset<circle::SubGraph>>; - using CircleBuffers_t = flatbuffers::Vector<flatbuffers::Offset<circle::Buffer>>; - using CircleTensors_t = flatbuffers::Vector<flatbuffers::Offset<circle::Tensor>>; - using CircleOperators_t = flatbuffers::Vector<flatbuffers::Offset<circle::Operator>>; - -public: - Reader(const circle::Model *model); - - Reader() = delete; - -public: - const std::vector<const circle::OperatorCode *> &opcodes() { return _op_codes; } - const CircleBuffers_t *buffers() { return _buffers; } - const CircleTensors_t *tensors() { return _tensors; } - const CircleOperators_t *operators() { return _operators; } - const std::vector<int32_t> &inputs() const { return _inputs; } - const std::vector<int32_t> &outputs() const { return _outputs; } - - uint32_t num_subgraph() const { return _subgraphs->Length(); } - - size_t buffer_info(uint32_t buf_idx, const uint8_t **buff_data); - circle::BuiltinOperator builtin_code(const circle::Operator *op) const; - std::string opcode_name(const circle::Operator *op) const; - -public: - bool select_subgraph(uint32_t subgraph); - -private: - const CircleSubGraphs_t *_subgraphs{nullptr}; - const CircleBuffers_t *_buffers{nullptr}; - const CircleTensors_t *_tensors{nullptr}; - const CircleOperators_t *_operators{nullptr}; - - std::vector<const circle::OperatorCode *> _op_codes; - std::vector<int32_t> _inputs; - std::vector<int32_t> _outputs; -}; - -} // namespace circletensordump - -#endif // __CIRCLE_TENSORDUMP_READER_H__ |