Make sure that flex delegate has TF core kernels through kernel_test
PiperOrigin-RevId: 329786213 Change-Id: I64760ae3c13a4fcb044a28ba0c636b84092d5bdf
This commit is contained in:
parent
6bac741e7a
commit
6128d6cee1
@ -177,10 +177,9 @@ tf_cc_test(
|
|||||||
srcs = ["kernel_test.cc"],
|
srcs = ["kernel_test.cc"],
|
||||||
tags = ["no_gpu"], # GPU + flex is not officially supported.
|
tags = ["no_gpu"], # GPU + flex is not officially supported.
|
||||||
deps = [
|
deps = [
|
||||||
|
":delegate",
|
||||||
":delegate_data",
|
":delegate_data",
|
||||||
":delegate_only_runtime",
|
|
||||||
":test_util",
|
":test_util",
|
||||||
"//tensorflow/core:all_kernels",
|
|
||||||
"@com_google_googletest//:gtest",
|
"@com_google_googletest//:gtest",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user