1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
// Copyright (c) Meta Platforms, Inc. and affiliates.
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.
#include <caffe2/torch/csrc/jit/backends/xnnpack/executor/xnn_executor.h>
#include <xnnpack.h>
#include <memory>
#include <vector>
namespace torch {
namespace jit {
namespace xnnpack {
namespace delegate {
class XNNCompiler {
public:
// Takes Flatbuffer Serialized XNNPack Model and rebuilds the xnn-subgraph
// returns an executor object that holds the xnn runtime object which we
// can then use to set inputs and run inference using the xnn graph.
static void compileModel(
const void* buffer_pointer,
size_t num_bytes,
XNNExecutor* executor);
};
} // namespace delegate
} // namespace xnnpack
} // namespace jit
} // namespace torch
|
