Merge pull request #855 from alibaba-damo-academy/dev_onnx

Xian Shi

2023-10-17 0dd01a880b8a7ed1d4549ee9b3a3d224e6fa1409

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#include "graphbuilderadapter.h"
 
#include "yaml-cpp/parser.h"  // IWYU pragma: keep
 
namespace YAML {
class GraphBuilderInterface;
 
void* BuildGraphOfNextDocument(Parser& parser,
                               GraphBuilderInterface& graphBuilder) {
  GraphBuilderAdapter eventHandler(graphBuilder);
  if (parser.HandleNextDocument(eventHandler)) {
    return eventHandler.RootNode();
  } else {
    return NULL;
  }
}
}