diff options
author | Brian Patton <bjp@google.com> | 2018-01-11 06:36:19 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-11 06:40:25 -0800 |
commit | 738dfa64cb2cc7771fa6bddb582abc8f32cff373 (patch) | |
tree | fc6145dac82bb1a5572cc3c38df45e2051149977 /tensorflow/compiler/xla/tests/test_utils.cc | |
parent | 198eca145f305fa35d9f3abd0e8261c30faa7fb8 (diff) |
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
Diffstat (limited to 'tensorflow/compiler/xla/tests/test_utils.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/test_utils.cc | 8 |
1 files changed, 1 insertions, 7 deletions
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 |