Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / src / include / index_select_inst.h

diff --git a/inference-engine/thirdparty/clDNN/src/include/index_select_inst.h b/inference-engine/thirdparty/clDNN/src/include/index_select_inst.h
index 0d775f6..b414849 100644 (file)
--- a/inference-engine/thirdparty/clDNN/src/include/index_select_inst.h
+++ b/inference-engine/thirdparty/clDNN/src/include/index_select_inst.h
@@ -36,7 +36,8 @@ namespace cldnn
                }
                program_node& input() const { return get_dependency(0); }
         program_node& indices() const { return get_dependency(1); }
-        index_select_axis_name get_axis() const { return get_primitive()->axis; }
+        bool get_reverse() const { return get_primitive()->reverse; }
+        std::vector<index_select_axis_name> get_axes() const { return get_primitive()->axis; }
        };
 
        using index_select_node = typed_program_node<index_select>;
@@ -53,7 +54,8 @@ namespace cldnn
 
         memory_impl& input() const { return dep_memory(0); }
         memory_impl& indices() const { return dep_memory(1); }
-        index_select_axis_name get_axis() const { return node.get_axis(); }
+        bool get_reverse() const { return node.get_reverse(); }
+        std::vector<index_select_axis_name> get_axes() const { return node.get_axes(); }
        };
 
        using index_select_inst = typed_primitive_inst<index_select>;