19 lines
325 B
Python
19 lines
325 B
Python
# Example Estimator model
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
py_binary(
|
|
name = "mnist_tflite",
|
|
srcs = [
|
|
"dataset.py",
|
|
"mnist_tflite.py",
|
|
],
|
|
python_version = "PY3",
|
|
deps = [
|
|
"//tensorflow:tensorflow_py",
|
|
],
|
|
)
|