Merge pull request tensorflow#12 from dgkutnic/dgkutnic/rn-i3d-updates

resnext passing, add annotations to i3d
plaidml · Aug 3, 2020 · d517a7b · d517a7b
2 parents 0621b55 + 0df7c2e
commit d517a7b
Show file tree

Hide file tree

Showing 6 changed files with 7,246 additions and 2,465 deletions.
diff --git a/tensorflow/compiler/xla/service/plaidml/tests/BUILD b/tensorflow/compiler/xla/service/plaidml/tests/BUILD
@@ -159,7 +159,7 @@ tf_cc_test(
 
 tf_cc_test(
     name = "plaidml_i3d_test",
-    srcs = ["plaidml_i3d_test.cc", "i3d_pretrained_inputs_and_weights.h"],
+    srcs = ["plaidml_i3d_test.cc", "i3d_pretrained_inputs_and_weights.h", "i3d_output.h"],
     extra_copts = ["-fexceptions"],
     deps = [
         "//tensorflow/compiler/xla/service:hlo",
@@ -273,6 +273,66 @@ tf_cc_test(
     ],
 )
 
+tf_cc_test(
+    name = "plaidml_resnext_stage2_test",
+    srcs = ["plaidml_resnext_stage2_test.cc", "resnext50_pretrained_inputs_and_weights.h", "resnext50_stage2_output.h"],
+    extra_copts = ["-fexceptions"],
+    deps = [
+        "//tensorflow/compiler/xla/service:hlo",
+        "//tensorflow/compiler/xla/service/plaidml:compiler",
+        "//tensorflow/compiler/xla/service/plaidml/tests:plaidml_codegen_test",
+        "//tensorflow/compiler/xla/tests:test_utils",
+        "//tensorflow/compiler/xla/tests:hlo_test_base",
+        "//tensorflow/core:framework",
+        "//tensorflow/core:lib",
+        "//tensorflow/core:test",
+        "//tensorflow/core:test_main",
+        "//tensorflow/core/platform:resource_loader",
+        "@com_google_absl//absl/strings",
+        "@com_google_googletest//:gtest",
+    ],
+)
+
+tf_cc_test(
+    name = "plaidml_resnext_stage3_test",
+    srcs = ["plaidml_resnext_stage3_test.cc", "resnext50_pretrained_inputs_and_weights.h", "resnext50_stage3_output.h"],
+    extra_copts = ["-fexceptions"],
+    deps = [
+        "//tensorflow/compiler/xla/service:hlo",
+        "//tensorflow/compiler/xla/service/plaidml:compiler",
+        "//tensorflow/compiler/xla/service/plaidml/tests:plaidml_codegen_test",
+        "//tensorflow/compiler/xla/tests:test_utils",
+        "//tensorflow/compiler/xla/tests:hlo_test_base",
+        "//tensorflow/core:framework",
+        "//tensorflow/core:lib",
+        "//tensorflow/core:test",
+        "//tensorflow/core:test_main",
+        "//tensorflow/core/platform:resource_loader",
+        "@com_google_absl//absl/strings",
+        "@com_google_googletest//:gtest",
+    ],
+)
+
+tf_cc_test(
+    name = "plaidml_resnext_stage4_test",
+    srcs = ["plaidml_resnext_stage4_test.cc", "resnext50_pretrained_inputs_and_weights.h", "resnext50_stage4_output.h"],
+    extra_copts = ["-fexceptions"],
+    deps = [
+        "//tensorflow/compiler/xla/service:hlo",
+        "//tensorflow/compiler/xla/service/plaidml:compiler",
+        "//tensorflow/compiler/xla/service/plaidml/tests:plaidml_codegen_test",
+        "//tensorflow/compiler/xla/tests:test_utils",
+        "//tensorflow/compiler/xla/tests:hlo_test_base",
+        "//tensorflow/core:framework",
+        "//tensorflow/core:lib",
+        "//tensorflow/core:test",
+        "//tensorflow/core:test_main",
+        "//tensorflow/core/platform:resource_loader",
+        "@com_google_absl//absl/strings",
+        "@com_google_googletest//:gtest",
+    ],
+)
+
 tf_cc_test(
     name = "plaidml_resnext_test_semantics",
     srcs = ["plaidml_resnext_test_semantics.cc"],