Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / src / inference_engine / net_pass.h

diff --git a/inference-engine/src/inference_engine/net_pass.h b/inference-engine/src/inference_engine/net_pass.h
index 8b19286..62e996f 100644 (file)
--- a/inference-engine/src/inference_engine/net_pass.h
+++ b/inference-engine/src/inference_engine/net_pass.h
@@ -1,4 +1,4 @@
-// Copyright (C) 2018 Intel Corporation
+// Copyright (C) 2018-2019 Intel Corporation
 // SPDX-License-Identifier: Apache-2.0
 //
 
@@ -15,17 +15,31 @@ namespace NetPass {
 
 /**
  * Try to detect LSTM Sequence pattern inside TI and convert it
+ *
  * @param net network to modify
  * @return true if all Tensor iterator was converted
  */
-INFERENCE_ENGINE_API_CPP(bool) CombineLSTMSeq(const ICNNNetwork &net);
+INFERENCE_ENGINE_API_CPP(bool) CombineRNNSeq(ICNNNetwork &net);
 
 /**
  * Unroll all present Tensor Iterators
+ *
  * @param net network to modify
  * @return true if all Tensor iterator was unrolled successfully
  */
-INFERENCE_ENGINE_API_CPP(bool) UnrollTI(const ICNNNetwork &net);
+INFERENCE_ENGINE_API_CPP(bool) UnrollTI(ICNNNetwork &net);
+
+/**
+ * Unroll all RNN specific layers by predicate
+ *
+ * Will be applied to all RNNSeq and RNNCell layers
+ *
+ * @param net network to modify
+ * @param pred predicate to mark layer to unroll
+ * @return true if all RNN layers was unrolled successfully
+ */
+INFERENCE_ENGINE_API_CPP(bool) UnrollRNN_if(ICNNNetwork &net,
+        std::function<bool(const RNNCellBase&)> pred);
 
 }  // namespace NetPass
 }  // namespace InferenceEngine