blob: 1ca33ae5411145fed2033cf4d01ff2b0c9346d4b (
plain)
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
|
DESCRIPTION = "TensorFlow Keras is an implementation of the Keras API that\
uses TensorFlow as a backend."
LICENSE = "Apache-2.0"
LIC_FILES_CHKSUM = "file://LICENSE;md5=175792518e4ac015ab6696d16c4f607e"
SRC_URI = "git://github.com/keras-team/tf-keras.git;branch=r2.17;protocol=https \
file://0001-customize-for-yocto.patch \
file://0001-skip-pip-file-check.patch \
file://0001-Keras-version-to-2.17.patch \
"
SRCREV = "9f64291e2db56f922917185ab8c1d5cd4e962021"
S = "${WORKDIR}/git"
inherit python3native bazel
DEPENDS += " \
python3-pip-native \
python3-wheel-native \
python3-absl-native \
python3-wrapt-native \
python3-opt-einsum-native \
python3-gast-native \
python3-astunparse-native \
python3-termcolor-native \
python3-typing-extensions-native \
python3-flatbuffers-native \
python3-mldtypes-native \
python3-pybind11-native \
python3-protobuf-native \
tensorflow-native \
"
do_compile () {
unset CC
export TMPDIR="${WORKDIR}"
export PYTHON_BIN_PATH="${PYTHON}"
${BAZEL} build \
--subcommands --explain=${T}/explain.log \
--verbose_explanations --verbose_failures \
--verbose_failures \
--python_path="${PYTHON}" \
//tf_keras/tools/pip_package:build_pip_package
${S}/bazel-bin/tf_keras/tools/pip_package/build_pip_package \
${WORKDIR}/tf_keras_pip
}
do_install () {
echo "Installing pip package"
install -d ${D}${PYTHON_SITEPACKAGES_DIR}
${STAGING_BINDIR_NATIVE}/pip3 install --disable-pip-version-check -v --no-deps \
-t ${D}/${PYTHON_SITEPACKAGES_DIR} --no-cache-dir ${WORKDIR}/tf_keras_pip/*.whl
rm -rf ${D}/${PYTHON_SITEPACKAGES_DIR}/tf_keras-${PV}.dist-info
# Provides module keras as usual
ln -snrf ${D}/${PYTHON_SITEPACKAGES_DIR}/tf_keras ${D}/${PYTHON_SITEPACKAGES_DIR}/keras
}
FILES:${PN} += "${libdir}/*"
BBCLASSEXTEND = "native"
|
