577 lines
22 KiB
Python
577 lines
22 KiB
Python
package(
|
|
default_visibility = ["//visibility:public"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
cc_library(
|
|
name = "add",
|
|
srcs = ["add.cc"],
|
|
hdrs = ["add.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "concat_xy",
|
|
srcs = ["concat_xy.cc"],
|
|
hdrs = ["concat_xy.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "concat_z",
|
|
srcs = ["concat_z.cc"],
|
|
hdrs = ["concat_z.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "conv_buffer_1x1",
|
|
srcs = ["conv_buffer_1x1.cc"],
|
|
hdrs = ["conv_buffer_1x1.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common:winograd_util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "conv_constants",
|
|
srcs = ["conv_constants.cc"],
|
|
hdrs = ["conv_constants.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "conv_powervr",
|
|
srcs = ["conv_powervr.cc"],
|
|
hdrs = ["conv_powervr.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common:winograd_util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "conv_weights_converter",
|
|
srcs = ["conv_weights_converter.cc"],
|
|
hdrs = ["conv_weights_converter.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "convolution_transposed",
|
|
srcs = ["convolution_transposed.cc"],
|
|
hdrs = ["convolution_transposed.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:storage_type_util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "convolution_transposed_3x3",
|
|
srcs = ["convolution_transposed_3x3.cc"],
|
|
hdrs = ["convolution_transposed_3x3.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "convolution_transposed_3x3_thin",
|
|
srcs = ["convolution_transposed_3x3_thin.cc"],
|
|
hdrs = ["convolution_transposed_3x3_thin.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "convolution_transposed_4x4",
|
|
srcs = ["convolution_transposed_4x4.cc"],
|
|
hdrs = ["convolution_transposed_4x4.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_conversion",
|
|
"//tensorflow/lite/delegates/gpu/common/task:weights_layout",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "convolution_transposed_thin",
|
|
srcs = ["convolution_transposed_thin.cc"],
|
|
hdrs = ["convolution_transposed_thin.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "depthwise_conv",
|
|
srcs = ["depthwise_conv.cc"],
|
|
hdrs = ["depthwise_conv.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "depthwise_conv_3x3",
|
|
srcs = ["depthwise_conv_3x3.cc"],
|
|
hdrs = ["depthwise_conv_3x3.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "elementwise",
|
|
srcs = ["elementwise.cc"],
|
|
hdrs = ["elementwise.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:storage_type_util",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "fully_connected",
|
|
srcs = ["fully_connected.cc"],
|
|
hdrs = ["fully_connected.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:buffer_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:texture2d_desc",
|
|
"@com_google_absl//absl/memory",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "lstm",
|
|
srcs = ["lstm.cc"],
|
|
hdrs = ["lstm.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "lstm_test_util",
|
|
testonly = 1,
|
|
srcs = ["lstm_test_util.cc"],
|
|
hdrs = ["lstm_test_util.h"],
|
|
deps = [
|
|
":lstm",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common/task:testing_util",
|
|
"@com_google_googletest//:gtest",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "max_unpooling",
|
|
srcs = ["max_unpooling.cc"],
|
|
hdrs = ["max_unpooling.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "mean_stddev_normalization",
|
|
srcs = ["mean_stddev_normalization.cc"],
|
|
hdrs = ["mean_stddev_normalization.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "padding",
|
|
srcs = ["padding.cc"],
|
|
hdrs = ["padding.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "pooling",
|
|
srcs = ["pooling.cc"],
|
|
hdrs = ["pooling.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "prelu",
|
|
srcs = ["prelu.cc"],
|
|
hdrs = ["prelu.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:storage_type_util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/types:variant",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "quantize_and_dequantize",
|
|
srcs = ["quantize_and_dequantize.cc"],
|
|
hdrs = ["quantize_and_dequantize.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:tensor",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/types:variant",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "reduce",
|
|
srcs = ["reduce.cc"],
|
|
hdrs = ["reduce.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:kernel_info",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "relu",
|
|
srcs = ["relu.cc"],
|
|
hdrs = ["relu.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "relu_test_util",
|
|
testonly = 1,
|
|
srcs = ["relu_test_util.cc"],
|
|
hdrs = ["relu_test_util.h"],
|
|
deps = [
|
|
":relu",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common/task:testing_util",
|
|
"@com_google_googletest//:gtest",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "reshape",
|
|
srcs = ["reshape.cc"],
|
|
hdrs = ["reshape.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "reshapex4",
|
|
srcs = ["reshapex4.cc"],
|
|
hdrs = ["reshapex4.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "softmax",
|
|
srcs = ["softmax.cc"],
|
|
hdrs = ["softmax.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "softmax1x1",
|
|
srcs = ["softmax1x1.cc"],
|
|
hdrs = ["softmax1x1.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:util",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "space_to_depth",
|
|
srcs = ["space_to_depth.cc"],
|
|
hdrs = ["space_to_depth.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "strided_slice",
|
|
srcs = ["strided_slice.cc"],
|
|
hdrs = ["strided_slice.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "transpose",
|
|
srcs = ["transpose.cc"],
|
|
hdrs = ["transpose.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "resize",
|
|
srcs = ["resize.cc"],
|
|
hdrs = ["resize.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:types",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "winograd",
|
|
srcs = ["winograd.cc"],
|
|
hdrs = ["winograd.h"],
|
|
deps = [
|
|
"//tensorflow/lite/delegates/gpu/common:data_type",
|
|
"//tensorflow/lite/delegates/gpu/common:operations",
|
|
"//tensorflow/lite/delegates/gpu/common:shape",
|
|
"//tensorflow/lite/delegates/gpu/common:status",
|
|
"//tensorflow/lite/delegates/gpu/common:winograd_util",
|
|
"//tensorflow/lite/delegates/gpu/common/task:gpu_operation",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:tensor_linear_desc",
|
|
"//tensorflow/lite/delegates/gpu/common/task:work_group_picking",
|
|
"@com_google_absl//absl/strings:str_format",
|
|
],
|
|
)
|