STT-tensorflow/tensorflow/lite/tools/optimize/sparsity/BUILD
A. Unique TensorFlower 0dfd23a641 Enable building TFLite Micro runtime for embedded systems under Bazel.
PiperOrigin-RevId: 305537308
Change-Id: Ia2a35cc06872709b27ca551316567edd4b241baa
2020-04-08 13:03:33 -07:00

35 lines
804 B
Python

load("//tensorflow/lite:build_def.bzl", "tflite_copts")
load("//tensorflow/lite/micro:build_def.bzl", "cc_library")
package(
default_visibility = [
"//visibility:public",
],
licenses = ["notice"], # Apache 2.0
)
cc_library(
name = "format_converter",
srcs = ["format_converter.cc"],
hdrs = ["format_converter.h"],
build_for_embedded = True,
copts = tflite_copts(),
deps = [
"//tensorflow/lite/c:common",
],
)
cc_test(
name = "format_converter_test",
srcs = ["format_converter_test.cc"],
data = ["//tensorflow/lite:testdata/sparse_tensor.bin"],
tags = [
"tflite_not_portable",
],
deps = [
":format_converter",
"//tensorflow/lite:framework",
"@com_google_googletest//:gtest_main",
],
)