wehub-resource-sync
1067 lines
32 KiB
Python
1067 lines
32 KiB
Python
load("@rules_cc//cc:cc_library.bzl", "cc_library")
|
|
load("@rules_cc//cc:cc_test.bzl", "cc_test")
|
|
|
|
# Implementation of StableHLO operations.
|
|
load("//tensorflow/lite/experimental/shlo:build_def.bzl", "shlo_ref_linkopts")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
|
|
default_visibility = ["//visibility:public"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "benchmark_util",
|
|
testonly = True,
|
|
hdrs = ["benchmark_util.h"],
|
|
deps = [
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "is_finite",
|
|
srcs = ["is_finite.cc"],
|
|
hdrs = ["is_finite.h"],
|
|
deps = [
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "is_finite_test",
|
|
srcs = ["is_finite_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":is_finite",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"//tensorflow/lite/experimental/shlo:tensor_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor_with_data",
|
|
"@com_google_absl//absl/types:span",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "is_finite_bench",
|
|
srcs = ["is_finite_bench.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":benchmark_util",
|
|
":is_finite",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"//tensorflow/lite/experimental/shlo:tensor_with_data",
|
|
"@com_google_absl//absl/log:absl_check",
|
|
"@com_google_benchmark//:benchmark",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "util",
|
|
srcs = ["util.cc"],
|
|
hdrs = ["util.h"],
|
|
deps = [
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "util_test",
|
|
srcs = ["util_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "unary_elementwise",
|
|
hdrs = ["unary_elementwise.h"],
|
|
deps = [
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/algorithm:container",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_absl//absl/types:span",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "unary_elementwise_test",
|
|
srcs = ["unary_elementwise_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":test_util",
|
|
":unary_elementwise",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/algorithm:container",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "test_util",
|
|
testonly = True,
|
|
hdrs = ["test_util.h"],
|
|
deps = [
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:i4",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/algorithm:container",
|
|
"@com_google_absl//absl/container:inlined_vector",
|
|
"@com_google_googletest//:gtest",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "unary_elementwise_test_util",
|
|
testonly = True,
|
|
hdrs = ["unary_elementwise_test_util.h"],
|
|
deps = [
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_googletest//:gtest",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "abs",
|
|
srcs = ["abs.cc"],
|
|
hdrs = ["abs.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "abs_test",
|
|
srcs = ["abs_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":abs",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "cbrt",
|
|
srcs = ["cbrt.cc"],
|
|
hdrs = ["cbrt.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "cbrt_test",
|
|
srcs = ["cbrt_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":cbrt",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "ceil",
|
|
srcs = ["ceil.cc"],
|
|
hdrs = ["ceil.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "ceil_test",
|
|
srcs = ["ceil_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":ceil",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "cosine",
|
|
srcs = ["cosine.cc"],
|
|
hdrs = ["cosine.h"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "cosine_test",
|
|
srcs = ["cosine_test.cc"],
|
|
deps = [
|
|
":cosine",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "sine",
|
|
srcs = ["sine.cc"],
|
|
hdrs = ["sine.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "sine_test",
|
|
srcs = ["sine_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":sine",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "tanh",
|
|
srcs = ["tanh.cc"],
|
|
hdrs = ["tanh.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "tanh_test",
|
|
srcs = ["tanh_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":tanh",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "sqrt",
|
|
srcs = ["sqrt.cc"],
|
|
hdrs = ["sqrt.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "sqrt_test",
|
|
srcs = ["sqrt_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":sqrt",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "exponential",
|
|
srcs = ["exponential.cc"],
|
|
hdrs = ["exponential.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "exponential_test",
|
|
srcs = ["exponential_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":exponential",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "floor",
|
|
srcs = ["floor.cc"],
|
|
hdrs = ["floor.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "floor_test",
|
|
srcs = ["floor_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":floor",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "log",
|
|
srcs = ["log.cc"],
|
|
hdrs = ["log.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "log_test",
|
|
srcs = ["log_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":log",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "exponential_minus_one",
|
|
srcs = ["exponential_minus_one.cc"],
|
|
hdrs = ["exponential_minus_one.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "exponential_minus_one_test",
|
|
srcs = ["exponential_minus_one_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":exponential_minus_one",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "log_plus_one",
|
|
srcs = ["log_plus_one.cc"],
|
|
hdrs = ["log_plus_one.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "log_plus_one_test",
|
|
srcs = ["log_plus_one_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":log_plus_one",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "logistic",
|
|
srcs = ["logistic.cc"],
|
|
hdrs = ["logistic.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "logistic_test",
|
|
srcs = ["logistic_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":logistic",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "sign",
|
|
srcs = ["sign.cc"],
|
|
hdrs = ["sign.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "sign_test",
|
|
srcs = ["sign_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":sign",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:bf16",
|
|
"//tensorflow/lite/experimental/shlo:f16",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "negate",
|
|
srcs = ["negate.cc"],
|
|
hdrs = ["negate.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "negate_test",
|
|
srcs = ["negate_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":negate",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "count_leading_zeros",
|
|
srcs = ["count_leading_zeros.cc"],
|
|
hdrs = ["count_leading_zeros.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:i4",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/numeric:bits",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "count_leading_zeros_test",
|
|
srcs = ["count_leading_zeros_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":count_leading_zeros",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:i4",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/numeric:bits",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "not",
|
|
srcs = ["not.cc"],
|
|
hdrs = ["not.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "not_test",
|
|
srcs = ["not_test.cc"],
|
|
deps = [
|
|
":not",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "popcnt",
|
|
srcs = ["popcnt.cc"],
|
|
hdrs = ["popcnt.h"],
|
|
deps = [
|
|
":unary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:i4",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/numeric:bits",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "popcnt_test",
|
|
srcs = ["popcnt_test.cc"],
|
|
deps = [
|
|
":popcnt",
|
|
":test_util",
|
|
":unary_elementwise_test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:i4",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/numeric:bits",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "binary_elementwise",
|
|
hdrs = ["binary_elementwise.h"],
|
|
deps = [
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "binary_elementwise_test",
|
|
srcs = ["binary_elementwise_test.cc"],
|
|
linkopts = shlo_ref_linkopts(),
|
|
deps = [
|
|
":binary_elementwise",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "binary_elementwise_test_util",
|
|
testonly = True,
|
|
hdrs = ["binary_elementwise_test_util.h"],
|
|
deps = [
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_googletest//:gtest",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "multiply",
|
|
srcs = ["multiply.cc"],
|
|
hdrs = ["multiply.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "multiply_test",
|
|
srcs = ["multiply_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":multiply",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "divide",
|
|
srcs = ["divide.cc"],
|
|
hdrs = ["divide.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "divide_test",
|
|
srcs = ["divide_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":divide",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "subtract",
|
|
srcs = ["subtract.cc"],
|
|
hdrs = ["subtract.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "subtract_test",
|
|
srcs = ["subtract_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":subtract",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "and",
|
|
srcs = ["and.cc"],
|
|
hdrs = ["and.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "and_test",
|
|
srcs = ["and_test.cc"],
|
|
deps = [
|
|
":and",
|
|
":binary_elementwise_test_util",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "or",
|
|
srcs = ["or.cc"],
|
|
hdrs = ["or.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "or_test",
|
|
srcs = ["or_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":or",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "xor",
|
|
srcs = ["xor.cc"],
|
|
hdrs = ["xor.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "xor_test",
|
|
srcs = ["xor_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":test_util",
|
|
":xor",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "maximum",
|
|
srcs = ["maximum.cc"],
|
|
hdrs = ["maximum.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "maximum_test",
|
|
srcs = ["maximum_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":maximum",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "minimum",
|
|
srcs = ["minimum.cc"],
|
|
hdrs = ["minimum.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "minimum_test",
|
|
srcs = ["minimum_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":minimum",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "compare",
|
|
srcs = ["compare.cc"],
|
|
hdrs = ["compare.h"],
|
|
deps = [
|
|
":binary_elementwise",
|
|
":util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:dispatch",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/status",
|
|
],
|
|
)
|
|
|
|
cc_test(
|
|
name = "compare_test",
|
|
srcs = ["compare_test.cc"],
|
|
deps = [
|
|
":binary_elementwise_test_util",
|
|
":compare",
|
|
":test_util",
|
|
"//tensorflow/lite/experimental/shlo:data_type",
|
|
"//tensorflow/lite/experimental/shlo:quantize",
|
|
"//tensorflow/lite/experimental/shlo:quantized_tensor_element_type",
|
|
"//tensorflow/lite/experimental/shlo:shape",
|
|
"//tensorflow/lite/experimental/shlo:status_matcher",
|
|
"//tensorflow/lite/experimental/shlo:tensor",
|
|
"@com_google_absl//absl/random",
|
|
"@com_google_absl//absl/random:distributions",
|
|
"@com_google_absl//absl/status",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|