Skip to content

Commit ef855a2

Browse files
dsharletgxnnpack-bot
authored andcommitted
Move operator API to internal header
This API is not intended to be or treated as a public API, we make frequent breaking changes to it, and plan to do more so in the future. PiperOrigin-RevId: 736176349
1 parent f74bbef commit ef855a2

36 files changed

+2302
-2268
lines changed

bench/operators/BUILD

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ OPERATOR_BENCHMARK_DEPS = [
2020
"//:common",
2121
"//:datatype",
2222
"//:math",
23+
"//:operators",
2324
]
2425

2526
xnnpack_benchmark(

bench/operators/average-pooling.cc

+1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
#include "include/xnnpack.h"
1919
#include "src/xnnpack/buffer.h"
2020
#include "src/xnnpack/common.h"
21+
#include "src/xnnpack/operator.h"
2122
#include <benchmark/benchmark.h>
2223
#ifdef BENCHMARK_TENSORFLOW_LITE
2324
#include <flatbuffers/include/flatbuffers/flatbuffers.h>

bench/operators/batch-matrix-multiply.cc

+1
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
#include "bench/utils.h"
1818
#include "include/xnnpack.h"
1919
#include "src/xnnpack/buffer.h"
20+
#include "src/xnnpack/operator.h"
2021
#include <benchmark/benchmark.h>
2122
#include <pthreadpool.h>
2223

bench/operators/binary.cc

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
#include "src/xnnpack/common.h"
2121
#include "src/xnnpack/datatype.h"
2222
#include "src/xnnpack/math.h"
23+
#include "src/xnnpack/operator.h"
2324
#include <benchmark/benchmark.h>
2425
#ifdef BENCHMARK_TENSORFLOW_LITE
2526
#include <flatbuffers/include/flatbuffers/buffer.h>

bench/operators/convolution.cc

+1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
#include "src/xnnpack/buffer.h"
2222
#include "src/xnnpack/common.h"
2323
#include "src/xnnpack/math.h"
24+
#include "src/xnnpack/operator.h"
2425
#include <benchmark/benchmark.h>
2526
#include <pthreadpool.h>
2627

bench/operators/deconvolution.cc

+1
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
#include "bench/utils.h"
1818
#include "include/xnnpack.h"
1919
#include "src/xnnpack/buffer.h"
20+
#include "src/xnnpack/operator.h"
2021
#include <benchmark/benchmark.h>
2122
#include <pthreadpool.h>
2223

bench/operators/fully-connected.cc

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
#include "include/xnnpack.h"
2121
#include "src/xnnpack/buffer.h"
2222
#include <benchmark/benchmark.h>
23+
#include "src/xnnpack/operator.h"
2324

2425
#define XNN_INVALID_NODE_ID UINT32_MAX
2526
void xnnpack_fully_connected_qd8_f32_qc4w(benchmark::State& state,

bench/operators/max-pooling.cc

+1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
#include "bench/utils.h"
1919
#include "include/xnnpack.h"
2020
#include "src/xnnpack/buffer.h"
21+
#include "src/xnnpack/operator.h"
2122
#include <benchmark/benchmark.h>
2223

2324
void max_pooling_u8(benchmark::State& state, const char* net) {

bench/operators/prelu.cc

+1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#include "bench/utils.h"
1616
#include "include/xnnpack.h"
1717
#include "src/xnnpack/buffer.h"
18+
#include "src/xnnpack/operator.h"
1819
#include <benchmark/benchmark.h>
1920
#ifdef BENCHMARK_TENSORFLOW_LITE
2021
#include <flatbuffers/include/flatbuffers/flatbuffers.h>

bench/operators/resize-bilinear-nhwc.cc

+1
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
#include "src/xnnpack/common.h"
1616
#include "src/xnnpack/datatype.h"
1717
#include "src/xnnpack/math.h"
18+
#include "src/xnnpack/operator.h"
1819
#include <benchmark/benchmark.h>
1920

2021
template <typename T>

bench/operators/softmax.cc

+1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
#include "include/xnnpack.h"
1919
#include "src/xnnpack/buffer.h"
2020
#include "src/xnnpack/math.h"
21+
#include "src/xnnpack/operator.h"
2122
#include <benchmark/benchmark.h>
2223

2324
#ifdef BENCHMARK_TENSORFLOW_LITE

bench/operators/unary.cc

+1
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
#include "src/xnnpack/common.h"
2121
#include "src/xnnpack/datatype.h"
2222
#include "src/xnnpack/math.h"
23+
#include "src/xnnpack/operator.h"
2324
#include <benchmark/benchmark.h>
2425
#ifdef BENCHMARK_TENSORFLOW_LITE
2526
#include <flatbuffers/include/flatbuffers/buffer.h>

0 commit comments

Comments
 (0)