experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
3742faa90c
STT-tensorflow/tensorflow/lite/kernels/BUILD
A. Unique TensorFlower 3742faa90c Support quantization to float16 
PiperOrigin-RevId: 248790891
2019-05-17 14:57:42 -07:00

1604 lines
37 KiB
Python
Raw Blame History

package(default_visibility = [
"//visibility:public",
])
licenses(["notice"]) # Apache 2.0
load("//tensorflow/lite:build_def.bzl", "tflite_copts")
load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite")
load("//tensorflow:tensorflow.bzl", "tf_opts_nortti_if_android")
# Enables usage of ruy in TF Lite kernels.
# WARNING: This build flag is experimental and subject to change.
config_setting(
name = "tflite_with_ruy",
define_values = {"tflite_with_ruy": "true"},
)
# Suppress warnings that are introduced by Eigen Tensor.
EXTRA_EIGEN_COPTS = select({
"//tensorflow:ios": [
"-Wno-error=invalid-partial-specialization",
"-Wno-error=reorder",
],
"//tensorflow:windows": [
"/DEIGEN_HAS_C99_MATH",
"/DEIGEN_AVOID_STL_ARRAY",
],
"//conditions:default": ["-Wno-error=reorder"],
})
cc_test(
name = "optional_tensor_test",
size = "small",
srcs = ["optional_tensor_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "test_util",
testonly = 1,
srcs = ["test_util.cc"],
hdrs = ["test_util.h"],
deps = [
":builtin_ops",
"//tensorflow/core:tflite_portable_logging",
"//tensorflow/lite:framework",
"//tensorflow/lite:schema_fbs_version",
"//tensorflow/lite:string_util",
"//tensorflow/lite/delegates/nnapi:nnapi_delegate",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"//tensorflow/lite/tools/optimize:quantization_utils",
"@com_google_googletest//:gtest",
],
)
# TODO(b/132204084): Update all kernel tests to use this main lib.
cc_library(
name = "test_main",
testonly = 1,
srcs = ["test_main.cc"],
deps = [
":test_util",
"//tensorflow/lite/testing:util",
"//tensorflow/lite/tools:command_line_flags",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "eigen_support",
srcs = [
"eigen_support.cc",
],
hdrs = [
"eigen_support.h",
],
copts = tflite_copts() + EXTRA_EIGEN_COPTS,
deps = [
":op_macros",
"//tensorflow/lite:arena_planner",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels/internal:optimized",
],
)
cc_test(
name = "eigen_support_test",
size = "small",
srcs = ["eigen_support_test.cc"],
deps = [
":eigen_support",
"//tensorflow/lite/kernels/internal:optimized",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "cpu_backend_context",
srcs = [
"cpu_backend_context.cc",
],
hdrs = [
"cpu_backend_context.h",
],
copts = tflite_copts(),
defines = select({
"//tensorflow/lite/kernels:tflite_with_ruy": [
"TFLITE_WITH_RUY",
],
"//conditions:default": [],
}),
deps = [
":op_macros",
# For now this unconditionally depends on both ruy and gemmlowp.
# See the comment inside class CpuBackendContext on the
# gemmlowp_context_ and ruy_context_ members.
"//tensorflow/lite/experimental/ruy:context",
"@gemmlowp",
],
)
cc_library(
name = "cpu_backend_threadpool",
hdrs = [
"cpu_backend_threadpool.h",
],
copts = tflite_copts(),
deps = [
"//tensorflow/lite/kernels/internal:types",
":cpu_backend_context",
# For now this unconditionally depends on both ruy and gemmlowp.
# We only need to depend on gemmlowp when tflite_with_ruy
# is false, but putting these dependencies in a select() seems to
# defeat copybara's rewriting rules.
"//tensorflow/lite/experimental/ruy:context",
"//tensorflow/lite/experimental/ruy:thread_pool",
"@gemmlowp",
],
)
cc_test(
name = "cpu_backend_threadpool_test",
srcs = ["cpu_backend_threadpool_test.cc"],
deps = [
":cpu_backend_context",
":cpu_backend_threadpool",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "cpu_backend_gemm",
srcs = [
"cpu_backend_gemm_ruy.h",
"cpu_backend_gemm_custom_gemv.h",
] + select({
"//tensorflow/lite/kernels:tflite_with_ruy": [],
"//conditions:default": [
"cpu_backend_gemm_gemmlowp.h",
"cpu_backend_gemm_eigen.h",
"cpu_backend_gemm_eigen.cc",
],
}),
hdrs = [
"cpu_backend_gemm.h",
"cpu_backend_gemm_params.h",
],
copts = tflite_copts(),
deps = [
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/kernels/internal:common",
":cpu_backend_context",
":cpu_backend_threadpool",
# Depend on ruy regardless of `tflite_with_ruy`. See the comment in
# cpu_backend_gemm.h about why ruy is the generic path.
"//tensorflow/lite/experimental/ruy",
# We only need to depend on gemmlowp and Eigen when tflite_with_ruy
# is false, but putting these dependencies in a select() seems to
# defeat copybara's rewriting rules.
"@gemmlowp",
"//third_party/eigen3",
],
)
cc_test(
name = "cpu_backend_gemm_test",
srcs = ["cpu_backend_gemm_test.cc"],
tags = ["notsan"],
deps = [
":cpu_backend_context",
":cpu_backend_gemm",
"@com_google_googletest//:gtest",
# ruy's reference path provides the reference implementation
# that this test compares against.
"//tensorflow/lite/experimental/ruy",
],
)
cc_library(
name = "cpu_backend_support",
srcs = [
"cpu_backend_support.cc",
],
hdrs = [
"cpu_backend_support.h",
],
copts = tflite_copts(),
deps = [
":cpu_backend_context",
":op_macros",
"//tensorflow/lite/c:c_api_internal",
"@gemmlowp",
],
)
cc_library(
name = "activation_functor",
hdrs = [
"activation_functor.h",
],
deps = [
"//tensorflow/lite/c:c_api_internal",
],
)
cc_library(
name = "op_macros",
hdrs = [
"op_macros.h",
],
)
cc_library(
name = "kernel_util",
srcs = [
"kernel_util.cc",
],
hdrs = [
"kernel_util.h",
],
deps = [
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels/internal:quantization_util",
"//tensorflow/lite/kernels/internal:round",
],
)
cc_test(
name = "kernel_util_test",
size = "small",
srcs = ["kernel_util_test.cc"],
deps = [
":kernel_util",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "test_util_test",
size = "small",
srcs = ["test_util_test.cc"],
deps = [
":test_util",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "padding",
srcs = [],
hdrs = ["padding.h"],
deps = [
"//tensorflow/lite/c:c_api_internal",
],
)
cc_library(
name = "builtin_op_kernels",
srcs = [
"activations.cc",
"add.cc",
"add_n.cc",
"arg_min_max.cc",
"audio_spectrogram.cc",
"basic_rnn.cc",
"batch_to_space_nd.cc",
"bidirectional_sequence_lstm.cc",
"bidirectional_sequence_rnn.cc",
"cast.cc",
"ceil.cc",
"comparisons.cc",
"concatenation.cc",
"conv.cc",
"depthwise_conv.cc",
"dequantize.cc",
"detection_postprocess.cc",
"div.cc",
"elementwise.cc",
"embedding_lookup.cc",
"embedding_lookup_sparse.cc",
"exp.cc",
"expand_dims.cc",
"fake_quant.cc",
"fill.cc",
"floor.cc",
"floor_div.cc",
"floor_mod.cc",
"fully_connected.cc",
"gather.cc",
"gather_nd.cc",
"hashtable_lookup.cc",
"if.cc",
"l2norm.cc",
"local_response_norm.cc",
"logical.cc",
"lsh_projection.cc",
"lstm.cc",
"matrix_diag.cc",
"matrix_set_diag.cc",
"maximum_minimum.cc",
"mfcc.cc",
"mirror_pad.cc",
"mul.cc",
"neg.cc",
"one_hot.cc",
"pack.cc",
"pad.cc",
"pooling.cc",
"pow.cc",
"quantize.cc",
"range.cc",
"rank.cc",
"reduce.cc",
"reshape.cc",
"resize_bilinear.cc",
"resize_nearest_neighbor.cc",
"reverse.cc",
"reverse_sequence.cc",
"round.cc",
"select.cc",
"shape.cc",
"skip_gram.cc",
"slice.cc",
"space_to_batch_nd.cc",
"space_to_depth.cc",
"sparse_to_dense.cc",
"split.cc",
"split_v.cc",
"squared_difference.cc",
"squeeze.cc",
"strided_slice.cc",
"sub.cc",
"svdf.cc",
"tile.cc",
"topk_v2.cc",
"transpose.cc",
"transpose_conv.cc",
"unidirectional_sequence_lstm.cc",
"unidirectional_sequence_rnn.cc",
"unique.cc",
"unpack.cc",
"where.cc",
"while.cc",
"zeros_like.cc",
],
hdrs = [
],
copts = tflite_copts() + tf_opts_nortti_if_android() + EXTRA_EIGEN_COPTS,
visibility = ["//visibility:private"],
deps = [
":activation_functor",
":cpu_backend_support",
":eigen_support",
":kernel_util",
":lstm_eval",
":op_macros",
":padding",
"//tensorflow/lite:framework",
"//tensorflow/lite:string_util",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels/internal:audio_utils",
"//tensorflow/lite/kernels/internal:kernel_utils",
"//tensorflow/lite/kernels/internal:optimized",
"//tensorflow/lite/kernels/internal:optimized_base",
"//tensorflow/lite/kernels/internal:quantization_util",
"//tensorflow/lite/kernels/internal:reference_base",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//third_party/eigen3",
"@farmhash_archive//:farmhash",
"@flatbuffers",
],
)
cc_library(
name = "lstm_eval",
srcs = ["lstm_eval.cc"],
hdrs = ["lstm_eval.h"],
deps = [
":kernel_util",
":op_macros",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels/internal:kernel_utils",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//third_party/eigen3",
"@gemmlowp",
],
)
cc_library(
name = "builtin_ops",
srcs = ["register.cc"],
hdrs = ["register.h"],
deps = [
":builtin_op_kernels",
"//tensorflow/lite:framework",
"//tensorflow/lite:util",
"//tensorflow/lite/c:c_api_internal",
],
)
# The builtin_ops target will resolve to optimized kernels when available. This
# target uses reference kernels only, and is useful for validation and testing.
# It should *not* generally be used in production.
cc_library(
name = "reference_ops",
srcs = ["register_ref.cc"],
hdrs = ["register_ref.h"],
deps = [
"//tensorflow/lite:framework",
"//tensorflow/lite:util",
"//tensorflow/lite/c:c_api_internal",
],
)
cc_test(
name = "audio_spectrogram_test",
size = "small",
srcs = ["audio_spectrogram_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "mfcc_test",
size = "small",
srcs = ["mfcc_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "detection_postprocess_test",
size = "small",
srcs = ["detection_postprocess_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "activations_test",
size = "small",
srcs = ["activations_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "add_test",
size = "small",
srcs = ["add_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "add_n_test",
size = "small",
srcs = ["add_n_test.cc"],
deps = [
":builtin_ops",
":test_util",
"//tensorflow/lite:framework",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "arg_min_max_test",
size = "small",
srcs = ["arg_min_max_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "div_test",
size = "small",
srcs = ["div_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "sub_test",
size = "small",
srcs = ["sub_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "transpose_test",
size = "small",
srcs = ["transpose_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"//tensorflow/lite/kernels/internal:reference",
"//tensorflow/lite/kernels/internal:reference_base",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "space_to_batch_nd_test",
size = "small",
srcs = ["space_to_batch_nd_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "batch_to_space_nd_test",
size = "small",
srcs = ["batch_to_space_nd_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "cast_test",
size = "small",
srcs = ["cast_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "concatenation_test",
size = "small",
srcs = ["concatenation_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "conv_test",
size = "small",
srcs = ["conv_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "depthwise_conv_test",
size = "small",
srcs = ["depthwise_conv_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "dequantize_test",
size = "small",
srcs = ["dequantize_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"//tensorflow/lite/kernels/internal:types",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "basic_rnn_test",
size = "small",
srcs = ["basic_rnn_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "bidirectional_sequence_lstm_test",
size = "small",
srcs = ["bidirectional_sequence_lstm_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "floor_test",
size = "small",
srcs = ["floor_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "ceil_test",
size = "small",
srcs = ["ceil_test.cc"],
tags = [
"tflite_not_portable_ios",
],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "round_test",
size = "small",
srcs = ["round_test.cc"],
tags = [
"tflite_not_portable_ios",
],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "elementwise_test",
size = "small",
srcs = ["elementwise_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unidirectional_sequence_lstm_test",
size = "small",
srcs = ["unidirectional_sequence_lstm_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "bidirectional_sequence_rnn_test",
size = "small",
srcs = ["bidirectional_sequence_rnn_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unidirectional_sequence_rnn_test",
size = "small",
srcs = ["unidirectional_sequence_rnn_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "l2norm_test",
size = "small",
srcs = ["l2norm_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "exp_test",
size = "small",
srcs = ["exp_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "fake_quant_test",
size = "small",
srcs = ["fake_quant_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "maximum_minimum_test",
size = "small",
srcs = ["maximum_minimum_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "reduce_test",
size = "small",
srcs = ["reduce_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "mul_test",
size = "small",
srcs = ["mul_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "pad_test",
size = "small",
srcs = ["pad_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "reshape_test",
size = "small",
srcs = ["reshape_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "gather_test",
size = "small",
srcs = ["gather_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "gather_nd_test",
size = "small",
srcs = ["gather_nd_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "topk_v2_test",
size = "small",
srcs = ["topk_v2_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "resize_bilinear_test",
size = "small",
srcs = ["resize_bilinear_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "resize_nearest_neighbor_test",
size = "small",
srcs = ["resize_nearest_neighbor_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "svdf_test",
size = "small",
srcs = ["svdf_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "embedding_lookup_test",
size = "small",
srcs = ["embedding_lookup_test.cc"],
deps = [
":builtin_ops",
":test_util",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels/internal:tensor",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "embedding_lookup_sparse_test",
size = "small",
srcs = ["embedding_lookup_sparse_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "fully_connected_test",
size = "small",
srcs = ["fully_connected_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"//tensorflow/lite/kernels/internal:tensor_utils",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "local_response_norm_test",
size = "small",
srcs = ["local_response_norm_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "pooling_test",
size = "small",
srcs = ["pooling_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "softmax_test",
size = "small",
srcs = ["softmax_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"//tensorflow/lite/kernels/internal:reference_base",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "log_softmax_test",
size = "small",
srcs = ["log_softmax_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"//tensorflow/lite/kernels/internal:reference_base",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "lsh_projection_test",
size = "small",
srcs = ["lsh_projection_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "hashtable_lookup_test",
size = "small",
srcs = ["hashtable_lookup_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite:string_util",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "lstm_test",
size = "small",
srcs = ["lstm_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "skip_gram_test",
size = "small",
srcs = ["skip_gram_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite:string_util",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "space_to_depth_test",
size = "small",
srcs = ["space_to_depth_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "split_test",
size = "small",
srcs = ["split_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "split_v_test",
size = "small",
srcs = ["split_v_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "squeeze_test",
size = "small",
srcs = ["squeeze_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "strided_slice_test",
size = "small",
srcs = ["strided_slice_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "tile_test",
size = "small",
srcs = ["tile_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "comparisons_test",
size = "small",
srcs = [
"comparisons_test.cc",
],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "neg_test",
size = "small",
srcs = ["neg_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "select_test",
size = "small",
srcs = [
"select_test.cc",
],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "slice_test",
size = "small",
srcs = [
"slice_test.cc",
],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "transpose_conv_test",
size = "small",
srcs = ["transpose_conv_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "expand_dims_test",
size = "small",
srcs = ["expand_dims_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "sparse_to_dense_test",
size = "small",
srcs = ["sparse_to_dense_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "shape_test",
size = "small",
srcs = ["shape_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "rank_test",
size = "small",
srcs = ["rank_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "pow_test",
size = "small",
srcs = ["pow_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "pack_test",
size = "small",
srcs = ["pack_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "one_hot_test",
size = "small",
srcs = ["one_hot_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "logical_test",
size = "small",
srcs = ["logical_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/c:c_api_internal",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unpack_test",
size = "small",
srcs = ["unpack_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "floor_div_test",
size = "small",
srcs = ["floor_div_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "where_test",
size = "small",
srcs = ["where_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "zeros_like_test",
size = "small",
srcs = ["zeros_like_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "floor_mod_test",
size = "small",
srcs = ["floor_mod_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "range_test",
size = "small",
srcs = ["range_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "squared_difference_test",
size = "small",
srcs = ["squared_difference_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "if_test",
size = "small",
srcs = ["if_test.cc"],
tags = ["tflite_not_portable_ios"],
deps = [
":builtin_ops",
":kernel_util",
":subgraph_test_util",
":test_util",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "while_test",
size = "small",
srcs = ["while_test.cc"],
tags = ["tflite_not_portable_ios"],
deps = [
":builtin_ops",
":kernel_util",
":subgraph_test_util",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "fill_test",
size = "small",
srcs = ["fill_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unique_test",
srcs = ["unique_test.cc"],
deps = [
":builtin_ops",
":test_util",
"//tensorflow/lite:framework",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "reverse_test",
size = "small",
srcs = ["reverse_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
filegroup(
name = "all_files",
srcs = glob(
["**/*"],
exclude = [
"**/METADATA",
"**/OWNERS",
],
),
visibility = ["//tensorflow:__subpackages__"],
)
tflite_portable_test_suite()
cc_test(
name = "mirror_pad_test",
srcs = ["mirror_pad_test.cc"],
deps = [
":builtin_ops",
":test_util",
"//tensorflow/lite:framework",
"@com_google_googletest//:gtest_main",
],
)
cc_library(
name = "subgraph_test_util",
testonly = 1,
srcs = ["subgraph_test_util.cc"],
hdrs = ["subgraph_test_util.h"],
deps = [
":builtin_ops",
":kernel_util",
":test_util",
"//tensorflow/lite:builtin_op_data",
"//tensorflow/lite:framework",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "subgraph_test_util_test",
size = "small",
srcs = ["subgraph_test_util_test.cc"],
deps = [
":kernel_util",
":subgraph_test_util",
":test_util",
"//tensorflow/lite:framework",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "reverse_sequence_test",
size = "small",
srcs = ["reverse_sequence_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "matrix_diag_test",
size = "small",
srcs = ["matrix_diag_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "quantize_test",
size = "small",
srcs = ["quantize_test.cc"],
deps = [
":builtin_ops",
":test_util",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels/internal:types",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "matrix_set_diag_test",
size = "small",
srcs = ["matrix_set_diag_test.cc"],
deps = [
":builtin_ops",
"//tensorflow/lite:framework",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest",
],
)
Reference in New Issue View Git Blame Copy Permalink