[XLA:DYNAMIC_PADDER] Do not pad init values into non-reduced dimensions

PiperOrigin-RevId: 431097193
cnuSODLearning · Feb 26, 2022 · 9e51184 · 9e51184
1 parent aa89283
commit 9e51184
Show file tree

Hide file tree

Showing 5 changed files with 203 additions and 188 deletions.
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD
@@ -3408,7 +3408,6 @@ cc_library(
         ":hlo_creation_utils",
         ":hlo_dce",
         ":hlo_pass",
-        ":hlo_verifier",
         ":pattern_matcher",
         ":shape_inference",
         "//tensorflow/compiler/xla:comparison_util",
@@ -3425,6 +3424,7 @@ cc_library(
         "@com_google_absl//absl/container:flat_hash_map",
         "@com_google_absl//absl/container:flat_hash_set",
         "@com_google_absl//absl/strings:str_format",
+        "@com_google_absl//absl/types:optional",
     ],
 )