123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555 |
- # Copyright 2020 The Chromium Authors. All rights reserved.
- # Use of this source code is governed by a BSD-style license that can be
- # found in the LICENSE file.
- import("//build/config/chromeos/ui_mode.gni")
- import("//build/config/sanitizers/sanitizers.gni")
- import("//third_party/protobuf/proto_library.gni")
- import("//third_party/tflite/features.gni")
- proto_library("tflite-config-proto") {
- proto_in_dir = "src"
- sources = [ "src/tensorflow/lite/experimental/acceleration/configuration/configuration.proto" ]
- cc_generator_options = "lite=true:"
- }
- config("tflite_config") {
- include_dirs = [ "src" ]
- if (is_android) {
- libs = [ "log" ]
- }
- }
- config("tflite_flags") {
- cflags = [
- "-Wno-c99-extensions",
- "-Wno-comment",
- "-Wno-extern-c-compat",
- "-Wno-gnu-inline-cpp-without-extern",
- "-Wno-loop-analysis",
- "-Wno-shadow",
- # TODO(crbug.com/1147556): Try to clean/remove no-sign-compare if possible.
- "-Wno-sign-compare",
- # TODO(thakis): Remove once
- # https://github.com/tensorflow/tensorflow/pull/50528 is rolled in.
- "-Wno-unreachable-code-break",
- "-Wno-unreachable-code-return",
- "-Wno-unused-but-set-variable",
- # This flag is needed to enable the strip error strings flag since a handful
- # of variables are only used in error logs, which are no-op'd in a macro
- # when the strings are stripped. Overall, removing the error strings saves
- # 90kb of binary size so having a few dangling variables is well worth it,
- # in addition to keeping the updating process simple.
- "-Wno-unused-variable",
- ]
- defines = [
- "FARMHASH_NO_CXX_STRING",
- "NEON2SSE_DISABLE_PERFORMANCE_WARNING",
- "TF_LITE_STRIP_ERROR_STRINGS",
- ]
- if (build_tflite_with_xnnpack) {
- defines += [ "TFLITE_BUILD_WITH_XNNPACK_DELEGATE" ]
- } else {
- defines += [ "TFLITE_WITHOUT_XNNPACK" ]
- }
- if (is_win) {
- cflags += [
- "/DTFL_COMPILE_LIBRARY",
- "/wd4018",
- ]
- }
- if (build_tflite_with_ruy) {
- defines += [ "TFLITE_WITH_RUY" ]
- } else if (is_linux || is_chromeos_lacros) {
- defines += [ "GEMMLOWP_ALLOW_SLOW_SCALAR_FALLBACK" ]
- }
- if (is_chromeos) {
- # TODO(crbug.com/1168787): Determine if neon can be supported
- # on chromeos builds.
- defines += [ "TF_LITE_DISABLE_X86_NEON" ]
- }
- }
- source_set("tflite_public_headers") {
- public = [
- "src/tensorflow/lite/allocation.h",
- "src/tensorflow/lite/arena_planner.h",
- "src/tensorflow/lite/c/builtin_op_data.h",
- "src/tensorflow/lite/c/common.h",
- "src/tensorflow/lite/context.h",
- "src/tensorflow/lite/context_util.h",
- "src/tensorflow/lite/core/macros.h",
- "src/tensorflow/lite/core/subgraph.h",
- "src/tensorflow/lite/error_reporter.h",
- "src/tensorflow/lite/graph_info.h",
- "src/tensorflow/lite/interpreter.h",
- "src/tensorflow/lite/interpreter_builder.h",
- "src/tensorflow/lite/kernels/builtin_op_kernels.h",
- "src/tensorflow/lite/model.h",
- "src/tensorflow/lite/model_builder.h",
- "src/tensorflow/lite/mutable_op_resolver.h",
- "src/tensorflow/lite/op_resolver.h",
- "src/tensorflow/lite/optional_debug_tools.h",
- "src/tensorflow/lite/schema/schema_generated.h",
- "src/tensorflow/lite/schema/schema_utils.h",
- "src/tensorflow/lite/signature_runner.h",
- "src/tensorflow/lite/stderr_reporter.h",
- "src/tensorflow/lite/string_type.h",
- "src/tensorflow/lite/string_util.h",
- "src/tensorflow/lite/util.h",
- ]
- configs += [ ":tflite_flags" ]
- }
- source_set("tflite_kernel_common_headers") {
- public = [
- "src/tensorflow/lite/experimental/resource/resource_variable.h",
- "src/tensorflow/lite/external_cpu_backend_context.h",
- "src/tensorflow/lite/kernels/cpu_backend_context.h",
- "src/tensorflow/lite/kernels/cpu_backend_gemm.h",
- "src/tensorflow/lite/kernels/cpu_backend_gemm_params.h",
- "src/tensorflow/lite/kernels/cpu_backend_threadpool.h",
- "src/tensorflow/lite/kernels/internal/compatibility.h",
- "src/tensorflow/lite/kernels/internal/tensor_ctypes.h",
- "src/tensorflow/lite/kernels/op_macros.h",
- ]
- configs += [ ":tflite_flags" ]
- visibility = [ ":*" ]
- }
- source_set("fft2d_deps") {
- public = [
- "src/third_party/fft2d/fft.h",
- "src/third_party/fft2d/fft2d.h",
- ]
- deps = [ "//third_party/fft2d" ]
- visibility = [ ":*" ]
- }
- source_set("tflite_kernel_internals") {
- sources = [
- "src/tensorflow/lite/kernels/internal/common.h",
- "src/tensorflow/lite/kernels/internal/cppmath.h",
- "src/tensorflow/lite/kernels/internal/kernel_utils.cc",
- "src/tensorflow/lite/kernels/internal/kernel_utils.h",
- "src/tensorflow/lite/kernels/internal/max.h",
- "src/tensorflow/lite/kernels/internal/min.h",
- "src/tensorflow/lite/kernels/internal/optimized/batch_matmul.h",
- "src/tensorflow/lite/kernels/internal/optimized/cpu_check.cc",
- "src/tensorflow/lite/kernels/internal/optimized/cpu_check.h",
- "src/tensorflow/lite/kernels/internal/optimized/depthwiseconv_3x3_filter_common.h",
- "src/tensorflow/lite/kernels/internal/optimized/depthwiseconv_float.h",
- "src/tensorflow/lite/kernels/internal/optimized/depthwiseconv_multithread.h",
- "src/tensorflow/lite/kernels/internal/optimized/depthwiseconv_uint8.h",
- "src/tensorflow/lite/kernels/internal/optimized/depthwiseconv_uint8_3x3_filter.h",
- "src/tensorflow/lite/kernels/internal/optimized/eigen_spatial_convolutions.h",
- "src/tensorflow/lite/kernels/internal/optimized/im2col_utils.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/add.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/conv.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/depthwise_conv.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/depthwise_conv_3x3_filter.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/depthwise_conv_hybrid.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/depthwise_conv_hybrid_3x3_filter.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/fully_connected.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/mean.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/mul.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/pooling.h",
- "src/tensorflow/lite/kernels/internal/optimized/integer_ops/transpose_conv.h",
- "src/tensorflow/lite/kernels/internal/optimized/multithreaded_conv.h",
- "src/tensorflow/lite/kernels/internal/optimized/neon_check.h",
- "src/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils.cc",
- "src/tensorflow/lite/kernels/internal/optimized/neon_tensor_utils_impl.h",
- "src/tensorflow/lite/kernels/internal/optimized/optimized_ops.h",
- "src/tensorflow/lite/kernels/internal/optimized/sparse_ops/fully_connected.h",
- "src/tensorflow/lite/kernels/internal/optimized/sse_tensor_utils.cc",
- "src/tensorflow/lite/kernels/internal/optimized/sse_tensor_utils_impl.h",
- "src/tensorflow/lite/kernels/internal/quantization_util.cc",
- "src/tensorflow/lite/kernels/internal/quantization_util.h",
- "src/tensorflow/lite/kernels/internal/reference/add.h",
- "src/tensorflow/lite/kernels/internal/reference/arg_min_max.h",
- "src/tensorflow/lite/kernels/internal/reference/batch_matmul.h",
- "src/tensorflow/lite/kernels/internal/reference/binary_function.h",
- "src/tensorflow/lite/kernels/internal/reference/ceil.h",
- "src/tensorflow/lite/kernels/internal/reference/comparisons.h",
- "src/tensorflow/lite/kernels/internal/reference/concatenation.h",
- "src/tensorflow/lite/kernels/internal/reference/conv.h",
- "src/tensorflow/lite/kernels/internal/reference/densify.h",
- "src/tensorflow/lite/kernels/internal/reference/depthwiseconv_float.h",
- "src/tensorflow/lite/kernels/internal/reference/depthwiseconv_uint8.h",
- "src/tensorflow/lite/kernels/internal/reference/dequantize.h",
- "src/tensorflow/lite/kernels/internal/reference/floor.h",
- "src/tensorflow/lite/kernels/internal/reference/fully_connected.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/add.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/conv.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/depthwise_conv.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/dequantize.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/fully_connected.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/l2normalization.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/log_softmax.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/logistic.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/mean.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/mul.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/pooling.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/tanh.h",
- "src/tensorflow/lite/kernels/internal/reference/integer_ops/transpose_conv.h",
- "src/tensorflow/lite/kernels/internal/reference/l2normalization.h",
- "src/tensorflow/lite/kernels/internal/reference/logistic.h",
- "src/tensorflow/lite/kernels/internal/reference/maximum_minimum.h",
- "src/tensorflow/lite/kernels/internal/reference/mul.h",
- "src/tensorflow/lite/kernels/internal/reference/neg.h",
- "src/tensorflow/lite/kernels/internal/reference/non_max_suppression.h",
- "src/tensorflow/lite/kernels/internal/reference/pad.h",
- "src/tensorflow/lite/kernels/internal/reference/pooling.h",
- "src/tensorflow/lite/kernels/internal/reference/portable_tensor_utils.cc",
- "src/tensorflow/lite/kernels/internal/reference/portable_tensor_utils.h",
- "src/tensorflow/lite/kernels/internal/reference/portable_tensor_utils_impl.h",
- "src/tensorflow/lite/kernels/internal/reference/prelu.h",
- "src/tensorflow/lite/kernels/internal/reference/process_broadcast_shapes.h",
- "src/tensorflow/lite/kernels/internal/reference/quantize.h",
- "src/tensorflow/lite/kernels/internal/reference/reduce.h",
- "src/tensorflow/lite/kernels/internal/reference/reference_ops.h",
- "src/tensorflow/lite/kernels/internal/reference/requantize.h",
- "src/tensorflow/lite/kernels/internal/reference/resize_nearest_neighbor.h",
- "src/tensorflow/lite/kernels/internal/reference/round.h",
- "src/tensorflow/lite/kernels/internal/reference/softmax.h",
- "src/tensorflow/lite/kernels/internal/reference/sparse_ops/fully_connected.h",
- "src/tensorflow/lite/kernels/internal/reference/strided_slice.h",
- "src/tensorflow/lite/kernels/internal/reference/sub.h",
- "src/tensorflow/lite/kernels/internal/reference/svdf.h",
- "src/tensorflow/lite/kernels/internal/reference/tanh.h",
- "src/tensorflow/lite/kernels/internal/spectrogram.cc",
- "src/tensorflow/lite/kernels/internal/spectrogram.h",
- "src/tensorflow/lite/kernels/internal/strided_slice_logic.h",
- "src/tensorflow/lite/kernels/internal/tensor.h",
- "src/tensorflow/lite/kernels/internal/tensor_utils.cc",
- "src/tensorflow/lite/kernels/internal/tensor_utils.h",
- "src/tensorflow/lite/kernels/internal/transpose_utils.cc",
- "src/tensorflow/lite/kernels/internal/transpose_utils.h",
- "src/tensorflow/lite/kernels/internal/types.h",
- "src/tensorflow/lite/kernels/internal/utils/sparsity_format_converter.cc",
- "src/tensorflow/lite/kernels/internal/utils/sparsity_format_converter.h",
- ]
- deps = [
- ":fft2d_deps",
- ":tflite_kernel_common_headers",
- ":tflite_public_headers",
- "//third_party/abseil-cpp:absl",
- "//third_party/farmhash",
- "//third_party/flatbuffers",
- "//third_party/fp16",
- "//third_party/gemmlowp",
- "//third_party/neon_2_sse",
- "//third_party/ruy",
- ]
- configs += [
- ":tflite_config",
- ":tflite_flags",
- "//third_party/eigen3:eigen_includes",
- ]
- visibility = [ ":*" ]
- }
- source_set("tflite_kernels") {
- sources = [
- "src/tensorflow/lite/kernels/activations.cc",
- "src/tensorflow/lite/kernels/add.cc",
- "src/tensorflow/lite/kernels/add_n.cc",
- "src/tensorflow/lite/kernels/arg_min_max.cc",
- "src/tensorflow/lite/kernels/assign_variable.cc",
- "src/tensorflow/lite/kernels/basic_rnn.cc",
- "src/tensorflow/lite/kernels/batch_matmul.cc",
- "src/tensorflow/lite/kernels/batch_to_space_nd.cc",
- "src/tensorflow/lite/kernels/bidirectional_sequence_lstm.cc",
- "src/tensorflow/lite/kernels/bidirectional_sequence_rnn.cc",
- "src/tensorflow/lite/kernels/builtin_op_kernels.h",
- "src/tensorflow/lite/kernels/cast.cc",
- "src/tensorflow/lite/kernels/ceil.cc",
- "src/tensorflow/lite/kernels/comparisons.cc",
- "src/tensorflow/lite/kernels/concatenation.cc",
- "src/tensorflow/lite/kernels/conv.cc",
- "src/tensorflow/lite/kernels/cpu_backend_context.cc",
- "src/tensorflow/lite/kernels/cpu_backend_gemm_custom_gemv.h",
- "src/tensorflow/lite/kernels/cpu_backend_gemm_eigen.cc",
- "src/tensorflow/lite/kernels/cpu_backend_gemm_eigen.h",
- "src/tensorflow/lite/kernels/cpu_backend_gemm_ruy.h",
- "src/tensorflow/lite/kernels/cumsum.cc",
- "src/tensorflow/lite/kernels/densify.cc",
- "src/tensorflow/lite/kernels/depth_to_space.cc",
- "src/tensorflow/lite/kernels/depthwise_conv.cc",
- "src/tensorflow/lite/kernels/dequantize.cc",
- "src/tensorflow/lite/kernels/dequantize.h",
- "src/tensorflow/lite/kernels/div.cc",
- "src/tensorflow/lite/kernels/eigen_support.cc",
- "src/tensorflow/lite/kernels/eigen_support.h",
- "src/tensorflow/lite/kernels/elementwise.cc",
- "src/tensorflow/lite/kernels/embedding_lookup.cc",
- "src/tensorflow/lite/kernels/embedding_lookup_sparse.cc",
- "src/tensorflow/lite/kernels/exp.cc",
- "src/tensorflow/lite/kernels/expand_dims.cc",
- "src/tensorflow/lite/kernels/fake_quant.cc",
- "src/tensorflow/lite/kernels/fill.cc",
- "src/tensorflow/lite/kernels/floor.cc",
- "src/tensorflow/lite/kernels/floor_div.cc",
- "src/tensorflow/lite/kernels/floor_mod.cc",
- "src/tensorflow/lite/kernels/fully_connected.cc",
- "src/tensorflow/lite/kernels/gather.cc",
- "src/tensorflow/lite/kernels/gather_nd.cc",
- "src/tensorflow/lite/kernels/hashtable_lookup.cc",
- "src/tensorflow/lite/kernels/if.cc",
- "src/tensorflow/lite/kernels/kernel_util.cc",
- "src/tensorflow/lite/kernels/kernel_util.h",
- "src/tensorflow/lite/kernels/l2norm.cc",
- "src/tensorflow/lite/kernels/local_response_norm.cc",
- "src/tensorflow/lite/kernels/logical.cc",
- "src/tensorflow/lite/kernels/lsh_projection.cc",
- "src/tensorflow/lite/kernels/lstm.cc",
- "src/tensorflow/lite/kernels/lstm_eval.cc",
- "src/tensorflow/lite/kernels/lstm_eval.h",
- "src/tensorflow/lite/kernels/lstm_shared.h",
- "src/tensorflow/lite/kernels/matrix_diag.cc",
- "src/tensorflow/lite/kernels/matrix_set_diag.cc",
- "src/tensorflow/lite/kernels/maximum_minimum.cc",
- "src/tensorflow/lite/kernels/mirror_pad.cc",
- "src/tensorflow/lite/kernels/mul.cc",
- "src/tensorflow/lite/kernels/neg.cc",
- "src/tensorflow/lite/kernels/non_max_suppression.cc",
- "src/tensorflow/lite/kernels/numeric_verify.cc",
- "src/tensorflow/lite/kernels/one_hot.cc",
- "src/tensorflow/lite/kernels/pack.cc",
- "src/tensorflow/lite/kernels/pad.cc",
- "src/tensorflow/lite/kernels/padding.h",
- "src/tensorflow/lite/kernels/pooling.cc",
- "src/tensorflow/lite/kernels/pow.cc",
- "src/tensorflow/lite/kernels/quantize.cc",
- "src/tensorflow/lite/kernels/range.cc",
- "src/tensorflow/lite/kernels/rank.cc",
- "src/tensorflow/lite/kernels/read_variable.cc",
- "src/tensorflow/lite/kernels/reduce.cc",
- "src/tensorflow/lite/kernels/reshape.cc",
- "src/tensorflow/lite/kernels/resize_bilinear.cc",
- "src/tensorflow/lite/kernels/resize_nearest_neighbor.cc",
- "src/tensorflow/lite/kernels/reverse.cc",
- "src/tensorflow/lite/kernels/reverse_sequence.cc",
- "src/tensorflow/lite/kernels/rfft2d.cc",
- "src/tensorflow/lite/kernels/round.cc",
- "src/tensorflow/lite/kernels/scatter_nd.cc",
- "src/tensorflow/lite/kernels/segment_sum.cc",
- "src/tensorflow/lite/kernels/select.cc",
- "src/tensorflow/lite/kernels/shape.cc",
- "src/tensorflow/lite/kernels/skip_gram.cc",
- "src/tensorflow/lite/kernels/slice.cc",
- "src/tensorflow/lite/kernels/space_to_batch_nd.cc",
- "src/tensorflow/lite/kernels/space_to_depth.cc",
- "src/tensorflow/lite/kernels/sparse_to_dense.cc",
- "src/tensorflow/lite/kernels/split.cc",
- "src/tensorflow/lite/kernels/split_v.cc",
- "src/tensorflow/lite/kernels/squared_difference.cc",
- "src/tensorflow/lite/kernels/squeeze.cc",
- "src/tensorflow/lite/kernels/strided_slice.cc",
- "src/tensorflow/lite/kernels/sub.cc",
- "src/tensorflow/lite/kernels/svdf.cc",
- "src/tensorflow/lite/kernels/tile.cc",
- "src/tensorflow/lite/kernels/topk_v2.cc",
- "src/tensorflow/lite/kernels/transpose.cc",
- "src/tensorflow/lite/kernels/transpose_conv.cc",
- "src/tensorflow/lite/kernels/unidirectional_sequence_lstm.cc",
- "src/tensorflow/lite/kernels/unidirectional_sequence_rnn.cc",
- "src/tensorflow/lite/kernels/unique.cc",
- "src/tensorflow/lite/kernels/unpack.cc",
- "src/tensorflow/lite/kernels/where.cc",
- "src/tensorflow/lite/kernels/while.cc",
- "src/tensorflow/lite/kernels/zeros_like.cc",
- ]
- if (!build_tflite_with_ruy) {
- sources += [ "src/tensorflow/lite/kernels/cpu_backend_gemm_gemmlowp.h" ]
- }
- deps = [
- ":fft2d_deps",
- ":tflite_kernel_common_headers",
- ":tflite_kernel_internals",
- ":tflite_public_headers",
- "//third_party/abseil-cpp:absl",
- "//third_party/farmhash",
- "//third_party/flatbuffers",
- "//third_party/fp16",
- "//third_party/gemmlowp",
- "//third_party/neon_2_sse",
- "//third_party/ruy",
- ]
- configs += [
- ":tflite_config",
- ":tflite_flags",
- "//third_party/eigen3:eigen_includes",
- ]
- visibility = [ ":*" ]
- }
- # TODO(crbug.com/1147996): Update to a component build. Will require updating the
- # tflite version.
- static_library("tflite") {
- sources = [
- "src/tensorflow/lite/allocation.cc",
- "src/tensorflow/lite/arena_planner.cc",
- "src/tensorflow/lite/builtin_ops.h",
- "src/tensorflow/lite/c/common.cc",
- "src/tensorflow/lite/c/common.h",
- "src/tensorflow/lite/core/api/error_reporter.cc",
- "src/tensorflow/lite/core/api/error_reporter.h",
- "src/tensorflow/lite/core/api/flatbuffer_conversions.cc",
- "src/tensorflow/lite/core/api/flatbuffer_conversions.h",
- "src/tensorflow/lite/core/api/op_resolver.cc",
- "src/tensorflow/lite/core/api/profiler.h",
- "src/tensorflow/lite/core/api/tensor_utils.cc",
- "src/tensorflow/lite/core/api/tensor_utils.h",
- "src/tensorflow/lite/core/subgraph.cc",
- "src/tensorflow/lite/delegates/interpreter_utils.cc",
- "src/tensorflow/lite/delegates/interpreter_utils.h",
- "src/tensorflow/lite/delegates/nnapi/nnapi_delegate.h",
- "src/tensorflow/lite/delegates/nnapi/nnapi_delegate_disabled.cc",
- "src/tensorflow/lite/experimental/acceleration/configuration/delegate_registry.cc",
- "src/tensorflow/lite/experimental/acceleration/configuration/delegate_registry.h",
- "src/tensorflow/lite/experimental/acceleration/configuration/flatbuffer_to_proto.cc",
- "src/tensorflow/lite/experimental/acceleration/configuration/flatbuffer_to_proto.h",
- "src/tensorflow/lite/experimental/acceleration/configuration/proto_to_flatbuffer.cc",
- "src/tensorflow/lite/experimental/acceleration/configuration/proto_to_flatbuffer.h",
- "src/tensorflow/lite/experimental/acceleration/mini_benchmark/mini_benchmark.cc",
- "src/tensorflow/lite/experimental/acceleration/mini_benchmark/mini_benchmark.h",
- "src/tensorflow/lite/experimental/resource/initialization_status.cc",
- "src/tensorflow/lite/experimental/resource/initialization_status.h",
- "src/tensorflow/lite/experimental/resource/lookup_interfaces.h",
- "src/tensorflow/lite/experimental/resource/lookup_util.h",
- "src/tensorflow/lite/experimental/resource/resource_base.h",
- "src/tensorflow/lite/experimental/resource/resource_variable.cc",
- "src/tensorflow/lite/experimental/resource/static_hashtable.cc",
- "src/tensorflow/lite/experimental/resource/static_hashtable.h",
- "src/tensorflow/lite/external_cpu_backend_context.cc",
- "src/tensorflow/lite/graph_info.cc",
- "src/tensorflow/lite/interpreter.cc",
- "src/tensorflow/lite/interpreter_builder.cc",
- "src/tensorflow/lite/interpreter_experimental.cc",
- "src/tensorflow/lite/memory_planner.h",
- "src/tensorflow/lite/minimal_logging.cc",
- "src/tensorflow/lite/minimal_logging.h",
- "src/tensorflow/lite/model.h",
- "src/tensorflow/lite/model_builder.cc",
- "src/tensorflow/lite/model_builder.h",
- "src/tensorflow/lite/mutable_op_resolver.cc",
- "src/tensorflow/lite/nnapi/NeuralNetworksTypes.h",
- "src/tensorflow/lite/nnapi/nnapi_implementation.h",
- "src/tensorflow/lite/nnapi/nnapi_implementation_disabled.cc",
- "src/tensorflow/lite/optional_debug_tools.cc",
- "src/tensorflow/lite/profiling/memory_info.cc",
- "src/tensorflow/lite/profiling/memory_info.h",
- "src/tensorflow/lite/profiling/platform_profiler.cc",
- "src/tensorflow/lite/profiling/platform_profiler.h",
- "src/tensorflow/lite/profiling/root_profiler.cc",
- "src/tensorflow/lite/profiling/root_profiler.h",
- "src/tensorflow/lite/profiling/time.cc",
- "src/tensorflow/lite/profiling/time.h",
- "src/tensorflow/lite/schema/schema_utils.cc",
- "src/tensorflow/lite/signature_runner.cc",
- "src/tensorflow/lite/simple_memory_arena.cc",
- "src/tensorflow/lite/simple_memory_arena.h",
- "src/tensorflow/lite/stderr_reporter.cc",
- "src/tensorflow/lite/stderr_reporter.h",
- "src/tensorflow/lite/string_util.cc",
- "src/tensorflow/lite/tflite_with_xnnpack_optional.cc",
- "src/tensorflow/lite/tflite_with_xnnpack_optional.h",
- "src/tensorflow/lite/tools/command_line_flags.h",
- "src/tensorflow/lite/tools/delegates/delegate_provider.h",
- "src/tensorflow/lite/tools/logging.h",
- "src/tensorflow/lite/tools/tool_params.h",
- "src/tensorflow/lite/tools/verifier.cc",
- "src/tensorflow/lite/tools/verifier_internal.cc",
- "src/tensorflow/lite/tools/verifier_internal.h",
- "src/tensorflow/lite/type_to_tflitetype.h",
- "src/tensorflow/lite/util.cc",
- "src/tensorflow/lite/version.h",
- ]
- if (build_tflite_with_xnnpack) {
- sources += [
- "src/tensorflow/lite/delegates/xnnpack/quantization_util.cc",
- "src/tensorflow/lite/delegates/xnnpack/quantization_util.h",
- "src/tensorflow/lite/delegates/xnnpack/xnnpack_delegate.cc",
- "src/tensorflow/lite/delegates/xnnpack/xnnpack_delegate.h",
- ]
- }
- if (is_win) {
- sources += [ "src/tensorflow/lite/mmap_allocation_disabled.cc" ]
- } else {
- sources += [ "src/tensorflow/lite/mmap_allocation.cc" ]
- }
- if (is_ios) {
- sources += [
- "src/tensorflow/lite/minimal_logging_ios.cc",
- "src/tensorflow/lite/profiling/signpost_profiler.h",
- "src/tensorflow/lite/profiling/signpost_profiler.mm",
- ]
- } else if (is_android) {
- sources += [
- "src/tensorflow/lite/minimal_logging_android.cc",
- "src/tensorflow/lite/profiling/atrace_profiler.cc",
- "src/tensorflow/lite/profiling/atrace_profiler.h",
- ]
- } else {
- sources += [ "src/tensorflow/lite/minimal_logging_default.cc" ]
- }
- deps = [
- "//third_party/abseil-cpp:absl",
- "//third_party/farmhash",
- "//third_party/fft2d",
- "//third_party/fp16",
- "//third_party/gemmlowp",
- "//third_party/neon_2_sse",
- "//third_party/ruy",
- ]
- if (build_tflite_with_xnnpack) {
- deps += [
- ":tflite_kernel_internals",
- "//third_party/cpuinfo",
- "//third_party/pthreadpool",
- "//third_party/xnnpack",
- ]
- }
- public_deps = [
- ":tflite-config-proto",
- ":tflite_kernel_common_headers",
- ":tflite_kernels",
- ":tflite_public_headers",
- "//third_party/flatbuffers",
- ]
- configs -= [ "//build/config/compiler:chromium_code" ]
- configs += [
- "//build/config/compiler:no_chromium_code",
- "//third_party/eigen3:eigen_includes",
- ":tflite_flags",
- ]
- public_configs = [ ":tflite_config" ]
- }
|