From 738dfa64cb2cc7771fa6bddb582abc8f32cff373 Mon Sep 17 00:00:00 2001
From: Brian Patton <bjp@google.com>
Date: Thu, 11 Jan 2018 06:36:19 -0800
Subject: Allow backends to specify a custom ShapeVerifier to HloVerifier.

Remove obsolete shape_size_fn_ from HloVerifier/ShapeVerifier.

Adds a rank check to FFT shape inference.

PiperOrigin-RevId: 181601294
---
 tensorflow/compiler/xla/tests/test_utils.cc | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

(limited to 'tensorflow/compiler/xla/tests/test_utils.cc')

diff --git a/tensorflow/compiler/xla/tests/test_utils.cc b/tensorflow/compiler/xla/tests/test_utils.cc
index bb215be8af..d7346d65c8 100644
--- a/tensorflow/compiler/xla/tests/test_utils.cc
+++ b/tensorflow/compiler/xla/tests/test_utils.cc
@@ -271,13 +271,7 @@ StatusOr<std::vector<std::unique_ptr<Literal>>> MakeFakeArguments(
 
 Status VerifyHloModule(const perftools::gputools::Platform& platform,
                        HloModule* const module) {
-  return HloVerifier(
-             std::bind(
-                 &TransferManager::GetByteSizeRequirement,
-                 TransferManager::GetForPlatform(&platform).ConsumeValueOrDie(),
-                 std::placeholders::_1))
-      .Run(module)
-      .status();
+  return HloVerifier().Run(module).status();
 }
 
 }  // namespace xla
-- 
cgit v1.2.3