blob: 1bee7f46d37ca249573357ecc8091e4f5c657440 (
plain) (
blame)
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
{ stdenv
, bash
, buildPackages
, buildBazelPackage
, fetchFromGitHub
, lib
}:
let
buildPlatform = stdenv.buildPlatform;
hostPlatform = stdenv.hostPlatform;
pythonEnv = buildPackages.python3.withPackages (ps: [ ps.numpy ]);
bazelDepsSha256ByBuildAndHost = {
x86_64-linux = {
x86_64-linux = "sha256-61qmnAB80syYhURWYJOiOnoGOtNa1pPkxfznrFScPAo=";
aarch64-linux = "sha256-sOIYpp98wJRz3RGvPasyNEJ05W29913Lsm+oi/aq/Ag=";
};
aarch64-linux = {
aarch64-linux = "sha256-MJU4y9Dt9xJWKgw7iKW+9Ur856rMIHeFD5u05s+Q7rQ=";
};
};
bazelHostConfigName.aarch64-linux = "elinux_aarch64";
bazelDepsSha256ByHost =
bazelDepsSha256ByBuildAndHost.${buildPlatform.system} or
(throw "unsupported build system ${buildPlatform.system}");
bazelDepsSha256 = bazelDepsSha256ByHost.${hostPlatform.system} or
(throw "unsupported host system ${hostPlatform.system} with build system ${buildPlatform.system}");
in
buildBazelPackage rec {
name = "tensorflow-lite";
version = "2.13.0";
src = fetchFromGitHub {
owner = "tensorflow";
repo = "tensorflow";
rev = "v${version}";
hash = "sha256-Rq5pAVmxlWBVnph20fkAwbfy+iuBNlfFy14poDPd5h0=";
};
bazel = buildPackages.bazel_5;
nativeBuildInputs = [ pythonEnv buildPackages.perl ];
bazelTargets = [
"//tensorflow/lite:libtensorflowlite.so"
"//tensorflow/lite/c:tensorflowlite_c"
"//tensorflow/lite/tools/benchmark:benchmark_model"
"//tensorflow/lite/tools/benchmark:benchmark_model_performance_options"
];
bazelFlags = [
"--config=opt"
] ++ lib.optionals (hostPlatform.system != buildPlatform.system) [
"--config=${bazelHostConfigName.${hostPlatform.system}}"
];
bazelBuildFlags = [ "--cxxopt=--std=c++17" ];
buildAttrs = {
installPhase = ''
mkdir -p $out/{bin,lib}
# copy the libs and binaries into the output dir
cp ./bazel-bin/tensorflow/lite/c/libtensorflowlite_c.so $out/lib
cp ./bazel-bin/tensorflow/lite/libtensorflowlite.so $out/lib
cp ./bazel-bin/tensorflow/lite/tools/benchmark/benchmark_model $out/bin
cp ./bazel-bin/tensorflow/lite/tools/benchmark/benchmark_model_performance_options $out/bin
find . -type f -name '*.h' | while read f; do
path="$out/include/''${f/.\//}"
install -D "$f" "$path"
# remove executable bit from headers
chmod -x "$path"
done
'';
};
fetchAttrs.sha256 = bazelDepsSha256;
PYTHON_BIN_PATH = pythonEnv.interpreter;
dontAddBazelOpts = true;
removeRulesCC = false;
postPatch = ''
rm .bazelversion
# Fix gcc-13 build failure by including missing include headers
sed -e '1i #include <cstdint>' -i \
tensorflow/lite/kernels/internal/spectrogram.cc
'';
preConfigure = ''
patchShebangs configure
'';
# configure script freaks out when parameters are passed
dontAddPrefix = true;
configurePlatforms = [];
meta = with lib; {
description = "An open source deep learning framework for on-device inference.";
homepage = "https://www.tensorflow.org/lite";
license = licenses.asl20;
maintainers = with maintainers; [ mschwaig cpcloud ];
platforms = [ "x86_64-linux" "aarch64-linux" ];
};
}
|