diff options
| author | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2019-09-16 15:48:19 +0900 |
|---|---|---|
| committer | GitHub Enterprise <noreply-CODE@samsung.com> | 2019-09-16 15:48:19 +0900 |
| commit | 56724606037d36b52a3ac7880809b2726a3d346a (patch) | |
| tree | 1462e624116601168373121083077c33c97411c4 /compiler | |
| parent | a3e4563d6bc4a80e3dfc45ddf1e0af140d071fd3 (diff) | |
| download | nnfw-56724606037d36b52a3ac7880809b2726a3d346a.tar.gz nnfw-56724606037d36b52a3ac7880809b2726a3d346a.tar.bz2 nnfw-56724606037d36b52a3ac7880809b2726a3d346a.zip | |
[exo-tflite] Handle export for TensorBroadcast (#7450)
This will add handler for loco::TensorBroadcast that throws an error
Signed-off-by: SaeHie Park <saehie.park@samsung.com>
Diffstat (limited to 'compiler')
| -rw-r--r-- | compiler/exo-tflite/src/OperationExporter.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/compiler/exo-tflite/src/OperationExporter.cpp b/compiler/exo-tflite/src/OperationExporter.cpp index 6b3400362..812aadf9e 100644 --- a/compiler/exo-tflite/src/OperationExporter.cpp +++ b/compiler/exo-tflite/src/OperationExporter.cpp @@ -90,6 +90,7 @@ public: void visit(loco::EltwiseDiv *) final; void visit(loco::EltwiseSqrt *) final; void visit(loco::FixedReshape *) final; + void visit(loco::TensorBroadcast *) final; void visit(locoex::COpCall *); @@ -662,6 +663,11 @@ void OperationExporter::visit(loco::FixedReshape *node) exportAsReshape(node, builder, new_shape_vec, gd); } +void OperationExporter::visit(loco::TensorBroadcast *) +{ + throw std::runtime_error("TensorBroadcast should not exist in the graph"); +} + inline flatbuffers::Offset<flatbuffers::Vector<uint8_t>> CreateCOpCallOptions(flatbuffers::FlatBufferBuilder &fbb, locoex::COpCall *copCall) { |