[nnkit/TF] TF backend working with zero input (#3303)

This commit adds TF backend working with zero input.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>

diff --git a/contrib/nnkit/libs/support/tf/src/Backend.cpp b/contrib/nnkit/libs/support/tf/src/Backend.cpp
index d534cf1..09a866e 100644 (file)
--- a/contrib/nnkit/libs/support/tf/src/Backend.cpp
+++ b/contrib/nnkit/libs/support/tf/src/Backend.cpp
@@ -48,8 +48,6 @@ Backend::Backend(const char *pb_path, const char *info_path) : _tf_runner(pb_pat
 
 void Backend::prepare(const std::function<void(nnkit::TensorContext &)> &f)
 {
-  assert(_inputs.size() > 0); // TODO support 0
-
   for (const auto &input_tensor : _inputs)
     _data_map.allocate(input_tensor.get());
 

diff --git a/contrib/nnkit/libs/support/tf/src/Runner.cpp b/contrib/nnkit/libs/support/tf/src/Runner.cpp
index 59fa755..416397d 100644 (file)
--- a/contrib/nnkit/libs/support/tf/src/Runner.cpp
+++ b/contrib/nnkit/libs/support/tf/src/Runner.cpp
@@ -166,7 +166,6 @@ Runner::~Runner()
 void Runner::prepareInputs(const std::vector<std::unique_ptr<ParsedTensor>> &inputs,
                            TensorDataMap &data_map)
 {
-  assert(inputs.size() > 0); // TODO support zero inputs
   assert(_graph);
 
   for (const auto &tensor : inputs)
@@ -217,9 +216,6 @@ void Runner::prepareOutputs(const std::vector<std::unique_ptr<ParsedTensor>> &ou
 void Runner::run()
 {
   assert(_graph);
-  assert(_input_ops.size() > 0); // TODO support with zero input
-  assert(_input_tensors.size() > 0);
-
   assert(_output_ops.size() > 0);
 
   TF_SessionOptions *options = TF_NewSessionOptions();