Google Git
Sign in
chromium / external / github.com / tensorflow / tensorflow.git / HEAD / . / tensorflow / lite / kernels / BUILD
blob: 9da18896842306cbee74d46c1dfdff40eb31ecec [file] [log] [blame]
load("@rules_cc//cc:cc_library.bzl", "cc_library")
load("@rules_cc//cc:cc_test.bzl", "cc_test")
load("//tensorflow:tensorflow.bzl", "tf_opts_nortti_if_android")
load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable", "pybind_extension")
load("//tensorflow/lite:build_def.bzl", "tflite_combine_cc_tests", "tflite_copts", "tflite_self_contained_libs_test_suite")
load("//tensorflow/lite:special_rules.bzl", "tflite_portable_test_suite_combined")
load("//tensorflow/lite/core/shims:cc_library_with_tflite.bzl", "cc_library_with_tflite")
package(
# copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
default_visibility = [
"//visibility:public",
],
licenses = ["notice"],
)
filegroup(
name = "tflite_internal_cc_3p_api_deps_src",
srcs = [
"builtin_ops_list.inc",
"op_macros.h",
],
visibility = [
"//tensorflow/lite:__pkg__",
],
)
# Enables usage of ruy exclusively as the GEMM backend in TFLite kernels.
# This will cause TFLite to build with ruy only, providing a smaller binary.
# WARNING: This build flag is experimental and subject to change.
config_setting(
name = "tflite_with_ruy_explicit_true",
define_values = {"tflite_with_ruy": "true"},
)
# Disables usage of ruy as the exclusive GEMM backend in TFLite kernels.
# TFLite will be built with ruy and other GEMM libraries. Ruy will not be
# the default GEMM option at runtime.
# WARNING: This build flag is experimental and subject to change.
config_setting(
name = "tflite_with_ruy_explicit_false",
define_values = {"tflite_with_ruy": "false"},
)
config_setting(
name = "aarch64",
constraint_values = [
"@platforms//cpu:aarch64",
],
visibility = ["//visibility:private"],
)
# Suppress warnings that are introduced by Eigen Tensor.
EXTRA_EIGEN_COPTS = select({
"//tensorflow:ios": [
"-Wno-error=invalid-partial-specialization",
"-Wno-error=reorder",
],
"//tensorflow:windows": [
# copybara:uncomment_begin(no MSVC flags in google)
# "-DEIGEN_HAS_C99_MATH",
# "-DEIGEN_AVOID_STL_ARRAY",
# copybara:uncomment_end_and_comment_begin
"/DEIGEN_HAS_C99_MATH",
"/DEIGEN_AVOID_STL_ARRAY",
# copybara:comment_end
],
"//conditions:default": ["-Wno-error=reorder"],
})
cc_test(
name = "optional_tensor_test",
size = "small",
srcs = ["optional_tensor_test.cc"],
deps = [
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest_main",
"@flatbuffers",
],
)
cc_library(
name = "acceleration_test_util",
testonly = 1,
srcs = [
"acceleration_test_util.cc",
],
hdrs = ["acceleration_test_util.h"],
deps = [
"@com_google_absl//absl/types:optional",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "acceleration_test_util_internal",
testonly = 1,
srcs = [
"acceleration_test_util_internal.cc",
],
hdrs = ["acceleration_test_util_internal.h"],
linkopts = select({
"//tensorflow:windows": [],
"//conditions:default": ["-lm"],
}),
deps = [
"@com_google_absl//absl/types:optional",
"@com_googlesource_code_re2//:re2",
],
)
cc_test(
name = "acceleration_test_util_internal_test",
srcs = [
"acceleration_test_util_internal_test.cc",
],
deps = [
":acceleration_test_util_internal",
"@com_google_googletest//:gtest_main",
],
)
cc_library(
name = "unidirectional_sequence_lstm_test_util",
testonly = 1,
hdrs = ["unidirectional_sequence_lstm_test_util.h"],
deps = [
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@flatbuffers",
],
)
cc_library(
name = "test_util",
testonly = 1,
srcs = ["test_util.cc"],
hdrs = ["test_util.h"],
deps = [
":acceleration_test_util",
":kernel_util",
":test_delegate_providers_lib",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:schema_fbs_version",
"//tensorflow/lite:simple_planner",
"//tensorflow/lite:string",
"//tensorflow/lite:string_util",
"//tensorflow/lite:type_to_tflitetype",
"//tensorflow/lite:util",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/api",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/core/kernels:builtin_ops",
"//tensorflow/lite/delegates/nnapi:acceleration_test_util",
"//tensorflow/lite/delegates/nnapi:nnapi_delegate",
"//tensorflow/lite/kernels/internal:tensor_ctypes",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//tensorflow/lite/kernels/internal/utils:sparsity_format_converter",
"//tensorflow/lite/nnapi:nnapi_implementation",
"//tensorflow/lite/schema:schema_conversion_utils",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"//tensorflow/lite/tools:logging",
"//tensorflow/lite/tools/optimize:quantization_utils",
"//tensorflow/lite/tools/serialization:writer_lib",
"//tensorflow/lite/tools/versioning",
"@FP16",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/base",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/base:no_destructor",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/strings:str_format",
"@com_google_absl//absl/synchronization",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
"@flatbuffers",
"@local_tsl//tsl/platform:logging",
],
)
# A convenient library of tflite delegate execution providers for kernel tests
# based on SingleOpModel or its derivatives defined in test_util.h/cc.
cc_library(
name = "test_util_delegate_providers",
copts = tflite_copts(),
deps = [
"//tensorflow/lite/tools/delegates:coreml_delegate_provider",
"//tensorflow/lite/tools/delegates:default_execution_provider",
"//tensorflow/lite/tools/delegates:delegate_provider_lib",
"//tensorflow/lite/tools/delegates:external_delegate_provider",
"//tensorflow/lite/tools/delegates:hexagon_delegate_provider",
"//tensorflow/lite/tools/delegates:nnapi_delegate_provider",
"//tensorflow/lite/tools/delegates:xnnpack_delegate_provider",
"//tensorflow/lite/tools/delegates/experimental/stable_delegate:delegate_provider",
] + select({
# Metal GPU delegate for iOS has its own setups for kernel tests, so
# skipping linking w/ the gpu_delegate_provider.
"//tensorflow:ios": [],
"//conditions:default": [
"//tensorflow/lite/tools/delegates:gpu_delegate_provider",
],
}),
alwayslink = 1,
# TODO(b/161243354): add testonly=1?
)
cc_library(
name = "test_delegate_providers_lib",
srcs = ["test_delegate_providers.cc"],
hdrs = ["test_delegate_providers.h"],
copts = tflite_copts(),
deps = [
"//tensorflow/lite/tools:command_line_flags",
"//tensorflow/lite/tools:logging",
"//tensorflow/lite/tools:tool_params",
"//tensorflow/lite/tools/delegates:delegate_provider_hdr",
"//tensorflow/lite/tools/delegates:delegate_provider_lib",
],
)
# TODO(b/132204084): Create tflite_cc_test rule to automate test_main inclusion.
cc_library(
name = "test_main",
testonly = 1,
srcs = ["test_main.cc"],
deps = [
":test_delegate_providers_lib",
":test_util",
":test_util_delegate_providers",
"//tensorflow/lite/testing:util",
"//tensorflow/lite/tools:command_line_flags",
"@com_google_benchmark//:benchmark",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "eigen_support",
srcs = [
"eigen_support.cc",
],
hdrs = [
"eigen_support.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts() + EXTRA_EIGEN_COPTS,
defines = ["EIGEN_NEON_GEBP_NR=4"],
visibility = ["//visibility:private"],
deps = [
":op_macros",
"//tensorflow/lite:util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:optimized_eigen",
],
)
cc_test(
name = "eigen_support_test",
size = "small",
srcs = ["eigen_support_test.cc"],
linkopts = select({
"//tensorflow:windows": [],
"//conditions:default": ["-lm"],
}),
deps = [
":eigen_support",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:optimized_eigen",
"@com_google_googletest//:gtest_main",
],
)
cc_library(
name = "tflite_with_ruy_enabled",
compatible_with = get_compatible_with_portable(),
defines = ["TFLITE_WITH_RUY"],
visibility = ["//visibility:private"],
)
cc_library(
name = "tflite_with_ruy_default",
compatible_with = get_compatible_with_portable(),
visibility = ["//visibility:private"],
deps = select({
":aarch64": [":tflite_with_ruy_enabled"],
"//tensorflow:android_arm": ["tflite_with_ruy_enabled"],
"//conditions:default": [],
}),
)
cc_library(
name = "tflite_with_ruy",
compatible_with = get_compatible_with_portable(),
deps = select({
":tflite_with_ruy_explicit_true": [":tflite_with_ruy_enabled"],
":tflite_with_ruy_explicit_false": [],
"//conditions:default": [":tflite_with_ruy_default"],
}),
)
# Provide a library for clients to link to if they need to stay on deprecated
# arithmetic backends. Include as a dependency of cpu_backend_gemm to start.
# TODO(b/168923364): Move to dependent targets.
cc_library(
name = "deprecated_backends",
srcs = [
"deprecated_backends.cc",
],
compatible_with = get_compatible_with_portable(),
alwayslink = 1,
)
cc_library(
name = "cpu_backend_context",
srcs = [
"cpu_backend_context.cc",
],
hdrs = [
"cpu_backend_context.h",
],
compatible_with = get_compatible_with_portable(),
# TF Lite builds in other build systems should "opt in" to cpufinfo.
copts = tflite_copts() + select({
"//tensorflow:linux_ppc64le": [],
"//tensorflow:linux_s390x": [],
"//tensorflow:fuchsia": [],
"//conditions:default": ["-DTFLITE_HAVE_CPUINFO"],
}),
defines = ["EIGEN_NEON_GEBP_NR=4"],
deps = [
":tflite_with_ruy",
":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.
"@ruy//ruy:context",
"@ruy//ruy:path",
"@gemmlowp",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite:macros",
"//tensorflow/lite:external_cpu_backend_context",
"//tensorflow/lite/kernels/internal:compatibility",
"@pthreadpool",
] + select({
# This select must match the similar select in `copts`
"//tensorflow:linux_ppc64le": [],
"//tensorflow:linux_s390x": [],
"//tensorflow:fuchsia": [],
"//conditions:default": ["@cpuinfo//:cpuinfo_with_unstripped_include_path"],
}),
)
cc_library(
name = "cpu_backend_threadpool",
hdrs = [
"cpu_backend_threadpool.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":cpu_backend_context",
":tflite_with_ruy",
"//tensorflow/lite/kernels/internal:compatibility",
# 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.
"@ruy//ruy:context",
"@ruy//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_main",
],
)
cc_library(
name = "cpu_backend_gemm",
srcs = [
"cpu_backend_gemm_custom_gemv.h",
"cpu_backend_gemm_eigen.cc",
"cpu_backend_gemm_eigen.h",
"cpu_backend_gemm_gemmlowp.h",
"cpu_backend_gemm_x86.h",
],
hdrs = [
"cpu_backend_gemm.h",
"cpu_backend_gemm_params.h",
"cpu_backend_gemm_ruy.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":tflite_with_ruy",
"//tensorflow/lite/kernels/internal:common",
"//tensorflow/lite/kernels/internal:compatibility",
"//tensorflow/lite/kernels/internal:cpu_check",
"//tensorflow/lite/kernels/internal:types",
":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.
"@ruy//ruy",
"@ruy//ruy:matrix",
"@ruy//ruy:path",
"@ruy//ruy:mul_params",
"@ruy//ruy/profiler:instrumentation",
# 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",
"@eigen_archive//:eigen3",
],
)
cc_test(
name = "cpu_backend_gemm_test",
srcs = ["cpu_backend_gemm_test.cc"],
deps = [
":cpu_backend_context",
":cpu_backend_gemm",
"@com_google_googletest//:gtest_main",
"@ruy//ruy:matrix",
# ruy:reference_mul provides the reference implementation
# that this test compares against.
"@ruy//ruy:reference_mul",
],
)
cc_library(
name = "op_macros",
hdrs = [
"op_macros.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
)
cc_library(
name = "stablehlo_elementwise",
srcs = ["stablehlo_elementwise.cc"],
hdrs = [
"stablehlo_elementwise.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":kernel_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:runtime_shape",
"//tensorflow/lite/kernels/internal:tensor_ctypes",
"//tensorflow/lite/kernels/internal:types",
"@eigen_archive//:eigen3",
],
)
cc_library(
name = "rng_util",
srcs = [
"rng_util.cc",
],
hdrs = [
"rng_util.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
)
cc_test(
name = "rng_util_test",
size = "small",
srcs = ["rng_util_test.cc"],
tags = ["tflite_smoke_test"],
deps = [
":rng_util",
":test_util",
"@com_google_googletest//:gtest_main",
],
)
cc_library(
name = "tensor_slice_util",
srcs = ["tensor_slice_util.cc"],
hdrs = [
"tensor_slice_util.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:runtime_shape",
"//tensorflow/lite/kernels/internal:tensor_ctypes",
],
)
cc_test(
name = "tensor_slice_util_test",
size = "small",
srcs = ["tensor_slice_util_test.cc"],
deps = [
":tensor_slice_util",
"//tensorflow/lite/core/c:c_api_types",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:runtime_shape",
"@com_google_googletest//:gtest_main",
],
)
cc_library(
name = "kernel_util",
srcs = [
"kernel_util.cc",
],
hdrs = [
"kernel_util.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":op_macros",
"//tensorflow/lite:array",
"//tensorflow/lite:kernel_api",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:cppmath",
"//tensorflow/lite/kernels/internal:quantization_util",
],
)
cc_test(
name = "kernel_util_test",
size = "small",
srcs = ["kernel_util_test.cc"],
linkopts = select({
"//tensorflow:windows": [],
"//conditions:default": ["-lm"],
}),
tags = ["tflite_smoke_test"],
deps = [
":builtin_ops", # build_cleaner: keep
":kernel_util",
":test_util",
"//tensorflow/lite:util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core:cc_api_stable",
"//tensorflow/lite/core/c:c_api_types",
"//tensorflow/lite/core/c:common",
"@com_google_absl//absl/strings",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "test_util_test",
size = "small",
srcs = ["test_util_test.cc"],
deps = [
":test_delegate_providers_lib",
":test_util",
"//tensorflow/lite:array",
"//tensorflow/lite:util",
"//tensorflow/lite/core/c:common",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "test_delegate_providers_lib_test",
size = "small",
srcs = ["test_delegate_providers_test.cc"],
# See details in https://github.com/bazelbuild/bazel/issues/11552 to avoid
# lazy symbol binding failure on macOS.
linkstatic = select({
"//tensorflow:macos": True,
"//conditions:default": False,
}),
deps = [
":test_delegate_providers_lib",
"//tensorflow/lite/tools/delegates:default_execution_provider",
"//tensorflow/lite/tools/delegates:nnapi_delegate_provider",
"//tensorflow/lite/tools/delegates:xnnpack_delegate_provider",
"@com_google_googletest//:gtest_main",
],
)
cc_library(
name = "padding",
srcs = [],
hdrs = ["padding.h"],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:types",
],
)
cc_library(
name = "control_flow_common",
srcs = ["control_flow_common.cc"],
hdrs = ["control_flow_common.h"],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":kernel_util",
"//tensorflow/lite:util",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
],
)
# See also VARIABLE_KERNEL_SRCS below.
BUILTIN_KERNEL_SRCS = [
"activations.cc",
"add.cc",
"add_n.cc",
"arg_min_max.cc",
"atan2.cc",
"audio_spectrogram.cc",
"basic_rnn.cc",
"batch_matmul.cc",
"batch_to_space_nd.cc",
"bidirectional_sequence_lstm.cc",
"bidirectional_sequence_rnn.cc",
"bitcast.cc",
"bitwise_xor.cc",
"broadcast_args.cc",
"broadcast_to.cc",
"bucketize.cc",
"call_once.cc",
"cast.cc",
"ceil.cc",
"comparisons.cc",
"complex_support.cc",
"concatenation.cc",
"conv.cc",
"conv3d.cc",
"conv3d_transpose.cc",
"cumsum.cc",
"densify.cc",
"depth_to_space.cc",
"depthwise_conv.cc",
"dequantize.cc",
"detection_postprocess.cc",
"dilate.cc",
"div.cc",
"dynamic_update_slice.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",
"hadamard_rotation.cc",
"hashtable.cc",
"hashtable_find.cc",
"hashtable_import.cc",
"hashtable_lookup.cc",
"hashtable_size.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",
"non_max_suppression.cc",
"numeric_verify.cc",
"one_hot.cc",
"pack.cc",
"pad.cc",
"pooling.cc",
"pow.cc",
"quantize.cc",
"random_ops.cc",
"range.cc",
"rank.cc",
"reduce.cc",
"reshape.cc",
"resize_bilinear.cc",
"resize_nearest_neighbor.cc",
"reverse.cc",
"reverse_sequence.cc",
"right_shift.cc",
"round.cc",
"scatter_nd.cc",
"segment_sum.cc",
"select.cc",
"shape.cc",
"stablehlo_gather.cc",
"stablehlo_add.cc",
"stablehlo_and.cc",
"stablehlo_composite.cc",
"stablehlo_case.cc",
"stablehlo_multiply.cc",
"stablehlo_pad.cc",
"stablehlo_reduce_window.cc",
"stablehlo_min_max.cc",
"stablehlo_scatter.cc",
"stablehlo_shift_left.cc",
"sign.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",
"unsorted_segment.cc",
"where.cc",
"while.cc",
"zeros_like.cc",
"rfft2d.cc",
"rng_bit_generator.cc",
]
VARIABLE_KERNEL_SRCS = [
"assign_variable.cc",
"read_variable.cc",
"var_handle.cc",
]
BUILTIN_KERNEL_DEPS = [
":cpu_backend_context",
":cpu_backend_gemm",
":cpu_backend_threadpool",
":kernel_util",
":rng_util",
":tensor_slice_util",
":lstm_eval",
":lstm_shared",
":op_macros",
":padding",
":stablehlo_elementwise",
":control_flow_common",
"@eigen_archive//:eigen3",
"@flatbuffers",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:minimal_logging",
"//tensorflow/lite:string_util",
"//tensorflow/lite:tflite_kernel_use_xnnpack_optional",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite:util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:audio_utils",
"//tensorflow/lite/kernels/internal:common",
"//tensorflow/lite/kernels/internal:compatibility",
"//tensorflow/lite/kernels/internal:cpu_check",
"//tensorflow/lite/kernels/internal:kernel_utils",
"//tensorflow/lite/kernels/internal:optimized_base",
"//tensorflow/lite/kernels/internal:portable_tensor_utils",
"//tensorflow/lite/kernels/internal:quantization_util",
"//tensorflow/lite/kernels/internal:reference_base",
"//tensorflow/lite/kernels/internal:strided_slice_logic",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//tensorflow/lite/kernels/internal:types",
] + select({
":tflite_with_ruy_explicit_true": [],
# Eigen multi-therading optimizations are only used when ruy is disabled.
"//conditions:default": [
":eigen_support",
"//tensorflow/lite/kernels/internal:optimized_eigen",
],
}) + select({
# This select must match the similar select in `copts`
"//tensorflow:linux_ppc64le": [],
"//tensorflow:linux_s390x": [],
"//tensorflow:fuchsia": [],
"//conditions:default": ["@cpuinfo//:cpuinfo_with_unstripped_include_path"],
})
cc_library(
name = "builtin_op_kernels",
srcs = BUILTIN_KERNEL_SRCS,
hdrs = [
"dequantize.h",
],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts() + tf_opts_nortti_if_android() + EXTRA_EIGEN_COPTS + select({
"//tensorflow:linux_ppc64le": [],
"//tensorflow:linux_s390x": [],
"//tensorflow:fuchsia": [],
"//conditions:default": ["-DTFLITE_HAVE_CPUINFO"],
}),
visibility = [
"//tensorflow/lite/core/kernels:__subpackages__",
],
deps = BUILTIN_KERNEL_DEPS + [
":variable_op_kernels",
"@FP16",
"@fft2d",
"@ruy//ruy/profiler:instrumentation",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite:array",
"//tensorflow/lite:builtin_ops",
"//tensorflow/lite:cc_api_stable",
"@local_xla//xla/tsl/lib/random:philox_random",
"@local_xla//xla/tsl/lib/random:random_distributions_utils",
"//tensorflow/lite/core/c:c_api_types",
# TODO(b/179298174): Move out from the experimental directory.
"//tensorflow/lite/experimental/resource",
"//tensorflow/lite/kernels/internal:cppmath",
"//tensorflow/lite:string",
"@farmhash_archive//:farmhash",
"//third_party/fft2d:fft2d_headers",
],
)
cc_library(
name = "variable_op_kernels",
srcs = VARIABLE_KERNEL_SRCS,
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":kernel_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/experimental/resource",
"//tensorflow/lite/kernels/internal:tensor",
"@flatbuffers",
],
)
cc_test(
name = "variable_ops_test",
size = "small",
srcs = [
"variable_ops_test.cc",
],
deps = [
":test_main",
":variable_op_kernels", # buildcleaner: keep
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/core/kernels:builtin_ops",
"//tensorflow/lite/kernels/internal:tensor",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
pybind_extension(
name = "pywrap_variable_ops",
srcs = [
"variable_ops_wrapper.cc",
],
additional_exported_symbols = ["VariableOpsRegisterer"],
enable_stub_generation = True,
link_in_framework = True,
pytype_srcs = [
"pywrap_variable_ops.pyi",
],
deps = [
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:mutable_op_resolver",
"@local_xla//third_party/python_runtime:headers",
"@pybind11",
],
)
cc_library(
name = "gru_cell",
srcs = ["gru_cell.cc"],
hdrs = ["gru_cell.h"],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
visibility = ["//visibility:private"],
deps = [
"//tensorflow/lite/kernels:cpu_backend_context",
"//tensorflow/lite/kernels/internal:optimized_base",
"//tensorflow/lite/kernels/internal:tensor",
"@eigen_archive//:eigen3",
],
)
cc_library(
name = "custom_ops",
srcs = [
"atan2_custom.cc",
"irfft2d.cc",
"multinomial.cc",
"pooling3d.cc",
"random_standard_normal_custom.cc",
"random_uniform_custom.cc",
"roll.cc",
"sign_custom.cc",
"table.cc",
"unidirectional_sequence_gru.cc",
],
hdrs = ["custom_ops_register.h"],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":gru_cell",
":kernel_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels:cpu_backend_context",
"//tensorflow/lite/kernels:padding",
"//tensorflow/lite/kernels/internal:common",
"//tensorflow/lite/kernels/internal:compatibility",
"//tensorflow/lite/kernels/internal:reference_base",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/kernels/internal:types",
"//third_party/fft2d:fft2d_headers",
"@fft2d",
"@flatbuffers",
"@ruy//ruy/profiler:instrumentation",
],
)
cc_library(
name = "lstm_eval",
srcs = ["lstm_eval.cc"],
hdrs = ["lstm_eval.h"],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":cpu_backend_context",
":op_macros",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:compatibility",
"//tensorflow/lite/kernels/internal:kernel_utils",
"//tensorflow/lite/kernels/internal:optimized_base",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/kernels/internal:tensor_utils",
"@ruy//ruy",
"@ruy//ruy/profiler:instrumentation",
],
)
cc_library(
name = "lstm_shared",
hdrs = ["lstm_shared.h"],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
)
cc_library(
name = "builtin_ops_list",
compatible_with = get_compatible_with_portable(),
textual_hdrs = ["builtin_ops_list.inc"],
)
exports_files(
srcs = ["builtin_ops_list.inc"],
visibility = ["//tensorflow/lite:__subpackages__"],
)
# For internal usage by shared libraries only.
cc_library_with_tflite(
name = "builtin_ops_all_linked",
hdrs = [
"builtin_op_kernels.h",
"fully_connected.h",
"register.h",
],
copts = tflite_copts(),
tflite_deps = [
"//tensorflow/lite:framework_stable",
],
# Limit visibility to TFLite only.
visibility = [
"//tensorflow/lite:__subpackages__",
],
deps = [
":builtin_op_kernels",
":builtin_ops_list",
"//tensorflow/lite:tflite_with_xnnpack_optional",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/core/kernels:builtin_ops_all_linked",
"//tensorflow/lite/schema:schema_fbs",
],
)
cc_library_with_tflite(
name = "builtin_ops",
hdrs = [
"builtin_op_kernels.h",
"fully_connected.h",
"register.h",
],
compatible_with = get_compatible_with_portable(),
generate_opaque_delegate_target = True,
tflite_deps = [
"//tensorflow/lite:cc_api_stable",
],
deps = [
":builtin_op_kernels", # buildcleaner: keep
":builtin_ops_list",
"//tensorflow/lite:mutable_op_resolver",
"//tensorflow/lite:tflite_with_xnnpack_optional",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/core/kernels:builtin_ops",
"//tensorflow/lite/schema:schema_fbs",
],
)
# 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"],
compatible_with = get_compatible_with_portable(),
copts = tflite_copts(),
deps = [
":builtin_op_kernels",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
],
)
cc_test(
name = "audio_spectrogram_test",
size = "small",
srcs = ["audio_spectrogram_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "mfcc_test",
size = "small",
srcs = ["mfcc_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "detection_postprocess_test",
size = "small",
srcs = ["detection_postprocess_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "activations_test",
size = "small",
srcs = ["activations_test.cc"],
tags = [
"tflite_nnapi",
"tflite_xnnpack",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/api",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "add_test",
size = "medium",
srcs = ["add_test.cc"],
shard_count = 10,
tags = [
"tflite_nnapi",
"tflite_xnnpack",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "add_n_test",
size = "small",
srcs = ["add_n_test.cc"],
deps = [
":add_n_test_common",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_library(
name = "add_n_test_common",
testonly = 1,
hdrs = [
"add_n_test_common.h",
],
deps = [
":test_util",
],
)
cc_test(
name = "arg_min_max_test",
size = "small",
srcs = ["arg_min_max_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "div_test",
size = "small",
srcs = ["div_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "sub_test",
size = "medium",
srcs = ["sub_test.cc"],
shard_count = 10,
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_library(
name = "transpose_test_utils",
hdrs = ["transpose_test_utils.h"],
deps = [
"//tensorflow/lite/kernels/internal:reference",
"//tensorflow/lite/kernels/internal:reference_base",
"//tensorflow/lite/kernels/internal:types",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/types:span",
],
)
cc_test(
name = "transpose_test_utils_test",
srcs = ["transpose_test_utils_test.cc"],
deps = [
":test_main",
":transpose_test_utils",
"@com_google_absl//absl/algorithm:container",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "transpose_test",
size = "small",
srcs = ["transpose_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
":transpose_test_utils",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/kernels/internal:compatibility",
"//tensorflow/lite/kernels/internal:reference",
"//tensorflow/lite/kernels/internal:reference_base",
"//tensorflow/lite/kernels/internal:tensor_utils_no_eigen",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "dilate_test",
size = "small",
srcs = ["dilate_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "reduce_window_test",
size = "small",
srcs = ["reduce_window_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "stablehlo_composite_test",
srcs = ["stablehlo_composite_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":kernel_util",
":subgraph_test_util",
":test_main",
":test_util",
"//tensorflow/lite:cc_api_stable",
"//tensorflow/lite/c:common",
"//tensorflow/lite/delegates/xnnpack:xnnpack_delegate_hdrs_only",
"//tensorflow/lite/kernels/internal:tensor_ctypes",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "stablehlo_case_test",
srcs = ["stablehlo_case_test.cc"],
deps = [
":kernel_util",
":subgraph_test_util",
":test_main",
":test_util",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core:cc_api_stable",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/log:absl_check",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "stablehlo_pad_test",
srcs = ["stablehlo_pad_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":stablehlo_reduce_window_test_util",
":test_main",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/random",
"@com_google_absl//absl/random:bit_gen_ref",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings:str_format",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "stablehlo_reduce_window_test_util",
hdrs = ["stablehlo_reduce_window_test_util.h"],
deps = [
"@com_google_absl//absl/algorithm:container",
],
)
cc_test(
name = "stablehlo_reduce_window_test_util_test",
size = "small",
srcs = ["stablehlo_reduce_window_test_util_test.cc"],
deps = [
":stablehlo_reduce_window_test_util",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "stablehlo_reduce_window_test",
srcs = ["stablehlo_reduce_window_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":stablehlo_reduce_window_test_util",
":subgraph_test_util",
":test_main",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/log:absl_log",
"@com_google_absl//absl/random",
"@com_google_absl//absl/random:bit_gen_ref",
"@com_google_absl//absl/random:distributions",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "space_to_batch_nd_test",
size = "small",
srcs = ["space_to_batch_nd_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "batch_to_space_nd_test",
size = "small",
srcs = ["batch_to_space_nd_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "batch_matmul_test",
size = "small",
srcs = ["batch_matmul_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "broadcast_args_test",
size = "small",
srcs = ["broadcast_args_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/kernels:builtin_ops",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "broadcast_to_test",
size = "small",
srcs = ["broadcast_to_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/kernels:builtin_ops",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "bucketize_test",
size = "small",
srcs = ["bucketize_test.cc"],
deps = [
":builtin_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "cast_test_common",
testonly = 1,
hdrs = [
"cast_test_common.h",
],
deps = [
":test_util",
"//tensorflow/lite:cc_api_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/types:span",
],
)
cc_test(
name = "cast_test",
size = "small",
srcs = ["cast_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":cast_test_common",
":test_main",
":test_util",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core/c:c_api_types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
],
)
cc_test(
name = "concatenation_test",
size = "small",
srcs = ["concatenation_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
"@flatbuffers",
],
)
cc_test(
name = "conv_test",
size = "small",
srcs = ["conv_test.cc"],
# TODO(b/162870360): Re-enable nnapi test after delegating grouped conv is added.
# tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "conv_mem_test",
srcs = ["conv_mem_test.cc"],
data = [
"//tensorflow/lite:testdata/conv3d_huge_im2col.bin",
"//tensorflow/lite:testdata/conv_huge_im2col.bin",
],
tags = [
"tflite_not_portable_ios",
"tflite_smoke_test",
],
deps = [
":kernel_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/core/kernels:builtin_ops",
"//tensorflow/lite/tools:logging",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "conv3d_test",
size = "small",
srcs = ["conv3d_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "densify_test",
size = "small",
srcs = ["densify_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "depthwise_conv_hybrid_test",
size = "small",
srcs = ["depthwise_conv_hybrid_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/api",
"//tensorflow/lite/kernels/internal:test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "depthwise_conv_test",
size = "small",
srcs = ["depthwise_conv_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/api",
"//tensorflow/lite/kernels/internal:test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "dequantize_test",
size = "small",
srcs = ["dequantize_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/api",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
"@flatbuffers",
],
)
cc_test(
name = "numeric_verify_test",
size = "small",
srcs = ["numeric_verify_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":kernel_util",
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/kernels/internal:reference",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
"@flatbuffers",
],
)
cc_test(
name = "basic_rnn_test",
size = "small",
srcs = ["basic_rnn_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "bidirectional_sequence_lstm_test",
size = "small",
srcs = ["bidirectional_sequence_lstm_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "floor_test",
size = "small",
srcs = ["floor_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "ceil_test",
size = "small",
srcs = ["ceil_test.cc"],
tags = [
"tflite_not_portable_ios",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "round_test",
size = "small",
srcs = ["round_test.cc"],
tags = [
"tflite_not_portable_ios",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
],
)
cc_test(
name = "elementwise_test",
size = "small",
srcs = ["elementwise_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:tensor_utils_no_eigen",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unidirectional_sequence_lstm_test",
size = "small",
srcs = ["unidirectional_sequence_lstm_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
":unidirectional_sequence_lstm_test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_benchmark//:benchmark",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "bidirectional_sequence_rnn_test",
size = "small",
srcs = ["bidirectional_sequence_rnn_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "unidirectional_sequence_rnn_test",
size = "small",
srcs = ["unidirectional_sequence_rnn_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "l2norm_test",
size = "small",
srcs = ["l2norm_test.cc"],
# TODO(b/143912164): Enable NNAPI test when fix nnapi.
# tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "exp_test",
size = "small",
srcs = ["exp_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "fake_quant_test",
size = "small",
srcs = ["fake_quant_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "maximum_minimum_test",
size = "small",
srcs = ["maximum_minimum_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
],
)
cc_library(
name = "reduce_test_common",
testonly = 1,
hdrs = [
"reduce_test_common.h",
],
deps = [
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/schema:schema_fbs",
],
)
cc_test(
name = "reduce_test",
size = "small",
srcs = ["reduce_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":reduce_test_common",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "mul_test",
size = "medium",
srcs = ["mul_test.cc"],
shard_count = 10,
tags = [
"tflite_nnapi",
"tflite_xnnpack",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "atan2_test",
size = "small",
srcs = ["atan2_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "atan2_custom_test",
size = "small",
srcs = ["atan2_custom_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "multinomial_test",
size = "small",
srcs = ["multinomial_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "roll_test",
size = "small",
srcs = ["roll_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "pad_test",
size = "small",
srcs = ["pad_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "random_ops_test",
size = "small",
srcs = ["random_ops_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/strings",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "random_standard_normal_custom_test",
size = "small",
srcs = ["random_standard_normal_custom_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "random_uniform_custom_test",
size = "small",
srcs = ["random_uniform_custom_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "sign_custom_test",
size = "small",
srcs = ["sign_custom_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "sign_test",
size = "small",
srcs = ["sign_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_library(
name = "reshape_test_common",
testonly = 1,
hdrs = [
"reshape_test_common.h",
],
deps = [
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/schema:schema_fbs",
],
)
cc_test(
name = "reshape_test",
size = "small",
srcs = ["reshape_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":reshape_test_common",
":test_main",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/core:framework_stable",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "gather_test",
size = "small",
srcs = ["gather_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "gather_nd_test",
size = "small",
srcs = ["gather_nd_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "scatter_nd_test",
size = "small",
srcs = ["scatter_nd_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "topk_v2_test",
size = "small",
srcs = ["topk_v2_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "resize_bilinear_test",
size = "small",
srcs = ["resize_bilinear_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/kernels/internal:optimized_base",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "resize_nearest_neighbor_test",
size = "small",
srcs = ["resize_nearest_neighbor_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "svdf_test",
size = "small",
srcs = ["svdf_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "embedding_lookup_test",
size = "small",
srcs = ["embedding_lookup_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "embedding_lookup_sparse_test",
size = "small",
srcs = ["embedding_lookup_sparse_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "fully_connected_test",
size = "small",
srcs = ["fully_connected_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":builtin_ops",
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/api",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "fully_connected_4bit_test",
size = "small",
srcs = ["fully_connected_4bit_test.cc"],
deps = [
":builtin_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "local_response_norm_test",
size = "small",
srcs = ["local_response_norm_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "pooling_test",
size = "small",
srcs = ["pooling_test.cc"],
tags = [
"tflite_nnapi",
"tflite_xnnpack",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "pooling3d_test",
size = "small",
srcs = ["pooling3d_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "softmax_test",
size = "small",
srcs = ["softmax_test.cc"],
tags = [
"tflite_nnapi",
"tflite_xnnpack",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/kernels/internal:reference_base",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "log_softmax_test",
size = "small",
srcs = ["log_softmax_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/kernels/internal:reference_base",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "lsh_projection_test",
size = "small",
srcs = ["lsh_projection_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "hashtable_lookup_test",
size = "small",
srcs = ["hashtable_lookup_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite:string_util",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "lstm_test",
size = "small",
srcs = ["lstm_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "lstm_eval_test",
size = "small",
srcs = ["lstm_eval_test.cc"],
deps = [
":cpu_backend_context",
":lstm_eval",
":test_main",
"//tensorflow/lite/core/c:common",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "skip_gram_test",
size = "small",
srcs = ["skip_gram_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite:string_util",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "space_to_depth_test",
size = "small",
srcs = ["space_to_depth_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "depth_to_space_test",
size = "small",
srcs = ["depth_to_space_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "split_test",
size = "small",
srcs = ["split_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "split_v_test",
size = "small",
srcs = ["split_v_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "squeeze_test",
size = "small",
srcs = ["squeeze_test.cc"],
tags = [
"tflite_nnapi",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "strided_slice_test",
size = "small",
srcs = ["strided_slice_test.cc"],
tags = [
"tflite_nnapi",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "table_test",
size = "small",
srcs = ["table_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite/kernels/internal:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "tile_test",
size = "small",
srcs = ["tile_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "comparisons_test",
size = "small",
srcs = [
"comparisons_test.cc",
],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "neg_test",
size = "small",
srcs = ["neg_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "select_test",
size = "small",
srcs = [
"select_test.cc",
],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "slice_test",
size = "small",
srcs = [
"slice_test.cc",
],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
],
)
cc_test(
name = "transpose_conv_test",
size = "small",
srcs = ["transpose_conv_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "expand_dims_test",
size = "small",
srcs = ["expand_dims_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "sparse_to_dense_test",
size = "small",
srcs = ["sparse_to_dense_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "shape_test",
size = "small",
srcs = ["shape_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "rank_test",
size = "small",
srcs = ["rank_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "pow_test",
size = "small",
srcs = ["pow_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/kernels/internal:test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "pack_test",
size = "small",
srcs = ["pack_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "one_hot_test",
size = "small",
srcs = ["one_hot_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "logical_test",
size = "small",
srcs = ["logical_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "unpack_test",
size = "small",
srcs = ["unpack_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "floor_div_test",
size = "small",
srcs = ["floor_div_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "where_test",
size = "small",
srcs = ["where_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:c_api_types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "zeros_like_test",
size = "small",
srcs = ["zeros_like_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_library(
name = "floor_mod_test_common",
testonly = 1,
hdrs = [
"floor_mod_test_common.h",
],
deps = [
":test_util",
],
)
cc_test(
name = "floor_mod_test",
size = "small",
srcs = ["floor_mod_test.cc"],
deps = [
":floor_mod_test_common",
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "range_test",
size = "small",
srcs = ["range_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "squared_difference_test",
size = "small",
srcs = ["squared_difference_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "call_once_test",
size = "small",
srcs = ["call_once_test.cc"],
tags = ["tflite_not_portable_ios"],
deps = [
":kernel_util",
":subgraph_test_util",
":test_main",
":variable_op_kernels",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "if_test",
size = "small",
srcs = ["if_test.cc"],
tags = ["tflite_not_portable_ios"],
deps = [
":kernel_util",
":subgraph_test_util",
":test_main",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/delegates/xnnpack:xnnpack_delegate",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/profiling:memory_info",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "while_test",
size = "small",
srcs = ["while_test.cc"],
tags = ["tflite_not_portable_ios"],
deps = [
":subgraph_test_util",
":test_main",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/delegates/xnnpack:xnnpack_delegate",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/profiling:memory_info",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "fill_test",
size = "small",
srcs = ["fill_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:string",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
],
)
cc_test(
name = "unique_test",
srcs = ["unique_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "reverse_test",
size = "small",
srcs = ["reverse_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "rfft2d_test",
size = "small",
srcs = ["rfft2d_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "irfft2d_test",
size = "small",
srcs = ["irfft2d_test.cc"],
deps = [
":custom_ops",
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "non_max_suppression_test",
size = "small",
srcs = ["non_max_suppression_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "mirror_pad_test",
srcs = ["mirror_pad_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_library(
name = "subgraph_test_util",
testonly = 1,
srcs = ["subgraph_test_util.cc"],
hdrs = ["subgraph_test_util.h"],
deps = [
":kernel_util",
"//tensorflow/lite:builtin_ops",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:interpreter_test_util",
"//tensorflow/lite:string_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/core/kernels:builtin_ops",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/algorithm:container",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "subgraph_test_util_test",
size = "small",
srcs = ["subgraph_test_util_test.cc"],
deps = [
":kernel_util",
":subgraph_test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "reverse_sequence_test",
size = "small",
srcs = ["reverse_sequence_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "matrix_diag_test",
size = "small",
srcs = ["matrix_diag_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "quantize_test",
size = "small",
srcs = ["quantize_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":kernel_util",
":test_main",
":test_util",
"//tensorflow/lite/kernels/internal:tensor_utils",
"//tensorflow/lite/kernels/internal:types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "matrix_set_diag_test",
size = "small",
srcs = ["matrix_set_diag_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "quant_basic_lstm_test",
size = "small",
srcs = ["quant_basic_lstm_test.cc"],
tags = ["tflite_nnapi"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "segment_sum_test",
srcs = ["segment_sum_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "complex_support_test",
srcs = ["complex_support_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "cumsum_test",
size = "small",
srcs = ["cumsum_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"//tensorflow/lite/testing:util",
"@com_google_googletest//:gtest",
],
)
exports_files(
[
"register.h",
"builtin_op_kernels.h",
"fully_connected.h",
],
visibility = ["//tensorflow/lite:__subpackages__"],
)
cc_test(
name = "hashtable_ops_test",
size = "small",
srcs = [
"hashtable_ops_test.cc",
],
deps = [
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/core/api",
"//tensorflow/lite/experimental/resource",
"//tensorflow/lite/kernels:test_main",
"//tensorflow/lite/kernels:test_util",
"//tensorflow/lite/kernels/internal:tensor",
"//tensorflow/lite/testing:util",
"@com_google_absl//absl/memory",
"@com_google_absl//absl/strings",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
cc_test(
name = "unidirectional_sequence_gru_test",
size = "small",
srcs = ["unidirectional_sequence_gru_test.cc"],
tags = ["tflite_not_portable_ios"],
deps = [
":custom_ops",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/kernels:test_util",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "conv3d_transpose_test",
size = "small",
srcs = ["conv3d_transpose_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite:string",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/memory",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "dynamic_update_slice_test",
size = "small",
srcs = ["dynamic_update_slice_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/kernels:subgraph_test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@eigen_archive//:eigen3",
"@flatbuffers",
],
)
cc_test(
name = "stablehlo_gather_test",
size = "small",
srcs = ["stablehlo_gather_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "stablehlo_add_test",
size = "small",
srcs = ["stablehlo_add_test.cc"],
deps = [
":subgraph_test_util",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "stablehlo_multiply_test",
size = "small",
srcs = ["stablehlo_multiply_test.cc"],
deps = [
":subgraph_test_util",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "stablehlo_and_test",
size = "small",
srcs = ["stablehlo_and_test.cc"],
deps = [
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "stablehlo_min_max_test",
size = "small",
srcs = ["stablehlo_min_max_test.cc"],
deps = [
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/c:common",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_absl//absl/log:absl_log",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "stablehlo_scatter_test",
size = "small",
srcs = ["stablehlo_scatter_test.cc"],
deps = [
":subgraph_test_util",
":test_main",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/core:subgraph",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "stablehlo_shift_left_test",
size = "small",
srcs = ["stablehlo_shift_left_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/c:c_api_types",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
# TODO(b/249321616) pull unsorted_segment_test* into seperate `cc_library`.
cc_test(
name = "unsorted_segment_prod_test",
size = "small",
srcs = [
"unsorted_segment_prod_test.cc",
"unsorted_segment_test.cc",
"unsorted_segment_test.h",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unsorted_segment_max_test",
size = "small",
srcs = [
"unsorted_segment_max_test.cc",
"unsorted_segment_test.cc",
"unsorted_segment_test.h",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unsorted_segment_sum_test",
size = "small",
srcs = [
"unsorted_segment_sum_test.cc",
"unsorted_segment_test.cc",
"unsorted_segment_test.h",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "unsorted_segment_min_test",
size = "small",
srcs = [
"unsorted_segment_sum_test.cc",
"unsorted_segment_test.cc",
"unsorted_segment_test.h",
],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/core/c:common",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "bitcast_test",
size = "small",
srcs = ["bitcast_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "bitwise_xor_test",
size = "small",
srcs = ["bitwise_xor_test.cc"],
deps = [
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "right_shift_test",
size = "small",
srcs = ["right_shift_test.cc"],
deps = [
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest_main",
],
)
cc_test(
name = "rng_bit_generator_test",
size = "small",
srcs = ["rng_bit_generator_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
],
)
cc_test(
name = "hadamard_rotation_test",
size = "small",
srcs = ["hadamard_rotation_test.cc"],
deps = [
":test_main",
":test_util",
"//tensorflow/lite:framework_stable",
"//tensorflow/lite/core:framework_stable",
"//tensorflow/lite/schema:schema_fbs",
"@com_google_googletest//:gtest",
"@flatbuffers",
],
)
tflite_portable_test_suite_combined(
combine_conditions = {"deps": [":test_main"]},
# TODO(b/229985981) : Remove `nnapi_args` after adding Relu0To1 is completed.
nnapi_args = ["--gunit_filter=*Reshape*"],
#nnapi_args = ["--gunit_filter=-*QuantizedActivationsOpTest.Relu0To1*"],
)
# Note this is created mainly for validating external delegates in OSS.
tflite_combine_cc_tests(
name = "combined_all_kernel_tests",
deps_conditions = [":test_main"],
generate_cc_library = True,
)
# This test uses SimpleAllocator and works nicely with ASAN.
tflite_combine_cc_tests(
name = "combined_all_kernel_with_simple_allocator_tests",
args = ["--use_simple_allocator=true"],
deps_conditions = [":test_main"],
)
tflite_self_contained_libs_test_suite(name = "self_contained_libs_tests")