Breakout get onnx op list in MIGraphX Cpp API

src/api/api.cpp

@@ -289,6 +289,13 @@ static void quantize_fp8_wrap(program& prog, const target& t, quantize_fp8_optio
     migraphx::quantize_fp8(prog, t, options.calibration);
 }
 
+static size_t get_onnx_operators_size() { return migraphx::get_onnx_operators().size(); }
+
+static const char* get_onnx_operator_name_at_index(std::size_t index)
+{
+    return get_onnx_operators().at(index).c_str();
+}
+
 #ifdef __clang__
 #pragma clang diagnostic push
 #pragma clang diagnostic ignored "-Wformat-nonliteral"
@@ -2416,6 +2423,19 @@ extern "C" migraphx_status migraphx_quantize_fp8(migraphx_program_t prog,
     return api_error_result;
 }
 
+extern "C" migraphx_status migraphx_get_onnx_operator_name_at_index(const char** out, size_t index)
+{
+    auto api_error_result =
+        migraphx::try_([&] { *out = migraphx::get_onnx_operator_name_at_index((index)); });
+    return api_error_result;
+}
+
+extern "C" migraphx_status migraphx_get_onnx_operators_size(size_t* out)
+{
+    auto api_error_result = migraphx::try_([&] { *out = migraphx::get_onnx_operators_size(); });
+    return api_error_result;
+}
+
 extern "C" migraphx_status migraphx_context_finish(const_migraphx_context_t context)
 {
     auto api_error_result = migraphx::try_([&] {

src/api/include/migraphx/migraphx.h

@@ -659,6 +659,11 @@ MIGRAPHX_C_EXPORT migraphx_status migraphx_quantize_fp8(migraphx_program_t prog,
                                                         migraphx_target_t target,
                                                         migraphx_quantize_fp8_options_t options);
 
+MIGRAPHX_C_EXPORT migraphx_status migraphx_get_onnx_operator_name_at_index(const char** out,
+                                                                           size_t index);
+
+MIGRAPHX_C_EXPORT migraphx_status migraphx_get_onnx_operators_size(size_t* out);
+
 MIGRAPHX_C_EXPORT migraphx_status migraphx_context_finish(const_migraphx_context_t context);
 
 MIGRAPHX_C_EXPORT migraphx_status migraphx_context_get_queue(void** out,

src/api/include/migraphx/migraphx.hpp

@@ -1460,7 +1460,7 @@
 };
 
 /// Parse a tf file into a migraphx program
-inline program parse_tf(const char* filename, const migraphx::tf_options& options)
+inline program parse_tf(const char* filename, const migraphx::tf_options& options->data())
 {
     return program(make<migraphx_program>(&migraphx_parse_tf, filename, options.get_handle_ptr()),
                    own{});
@@ -1643,6 +1643,20 @@
     op.register_op();
 }
 
+std::vector<std::string> get_onnx_operators()
+{
+    auto size = get_onnx_operators_size();
+    std::vector<std::string> result(size, "");
+
+    size_t index = 0;
+    for(auto &name: result)
+    {
+        name = get_onnx_operator_name_at_index(index);
+        index++;
+    }
+    return result;
+}
+
 #ifndef DOXYGEN
 } // namespace api
 #endif

src/api/migraphx.py

@@ -510,6 +510,16 @@ def quantize_fp8_options(h):
                             options='migraphx::quantize_fp8_options'),
                  fname='migraphx::quantize_fp8_wrap')
 
+api.add_function('migraphx_get_onnx_operator_name_at_index',
+                 api.params(index='size_t'),
+                 fname='migraphx::get_onnx_operator_name_at_index',
+                 returns='const char *')
+
+api.add_function('migraphx_get_onnx_operators_size',
+                 fname='migraphx::get_onnx_operators_size',
+                 returns='size_t')
+
+
 
 @auto_handle(ref=True)
 def context(h):

src/include/migraphx/onnx.hpp

@@ -76,7 +76,7 @@ MIGRAPHX_ONNX_EXPORT program parse_onnx_buffer(const void* data,
                                                std::size_t size,
                                                const onnx_options& options);
 
-MIGRAPHX_ONNX_EXPORT std::vector<std::string> get_onnx_operators();
+MIGRAPHX_ONNX_EXPORT const std::vector<std::string>& get_onnx_operators();
 
 } // namespace MIGRAPHX_INLINE_NS
 } // namespace migraphx

src/onnx/onnx.cpp

@@ -109,7 +109,11 @@ program parse_onnx_buffer(const void* data, std::size_t size, const onnx_options
     return parse_onnx_from(options, data, size);
 }
 
-std::vector<std::string> get_onnx_operators() { return onnx::get_op_parsers(); }
+const std::vector<std::string>& get_onnx_operators() 
+{ 
+    static std::vector<std::string> result = onnx::get_op_parsers();
+    return result;
+}
 
 } // namespace MIGRAPHX_INLINE_NS
 } // namespace migraphx

src/py/migraphx_py.cpp

@@ -621,6 +621,7 @@ MIGRAPHX_PYBIND11_MODULE(migraphx, m)
         py::arg("map_input_dims") = std::unordered_map<std::string, std::vector<std::size_t>>(),
         py::arg("output_names")   = std::vector<std::string>());
 
+    m.def("get_onnx_operators", []{return migraphx::get_onnx_operators();});
     m.def(
         "parse_onnx",
         [](const std::string& filename,

tools/api/api.cpp

@@ -289,6 +289,16 @@ static void quantize_fp8_wrap(program& prog, const target& t, quantize_fp8_optio
     migraphx::quantize_fp8(prog, t, options.calibration);
 }
 
+static size_t get_onnx_operators_size()
+{
+    return migraphx::get_onnx_operators().size();
+}
+
+static const char * get_onnx_operator_name_at_index(std::size_t index)
+{
+    return get_onnx_operators().at(index).c_str();   
+}
+
 #ifdef __clang__
 #pragma clang diagnostic push
 #pragma clang diagnostic ignored "-Wformat-nonliteral"