diff options
author | Eli Bendersky <eliben@google.com> | 2017-05-18 13:13:38 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-18 13:17:14 -0700 |
commit | e84588a3e9e67d91d6ae32e64469f890d217c5dd (patch) | |
tree | fab8672a31d1e2e7477aa1ff197bb3098fe516f5 /tensorflow/compiler/xla/tools | |
parent | 315f195f728d59647f23f10ecc664b8ce9466a06 (diff) |
[XLA] Attach an HloModuleConfig to HloModule, obviating the need to pass them around as a pair.
This cuts through a bunch of critical XLA APIs, but it's time... The background for this change is to make flags/options more easily pipe-able from the TF/XLA boundary deep into the XLA compiler and other components.
The situation after this CL is still not perfect; there are a number of places with chicken-egg scenarios when a module has to be constructed before a config (to register the result shape), but the situation is strictly better than before. Future CLs will clean things up even more.
PiperOrigin-RevId: 156469639
Diffstat (limited to 'tensorflow/compiler/xla/tools')
-rw-r--r-- | tensorflow/compiler/xla/tools/dumped_computation_to_text.cc | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc b/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc index 1f0ca31d6d..dc5a86f34e 100644 --- a/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc +++ b/tensorflow/compiler/xla/tools/dumped_computation_to_text.cc @@ -76,7 +76,10 @@ void RealMain(tensorflow::gtl::ArraySlice<char*> args, bool compile) { VersionedComputationHandle versioned_handle = user_computation->GetVersionedHandle(); std::unique_ptr<HloModule> module = - tracker.BuildHloModule(versioned_handle).ConsumeValueOrDie(); + tracker + .BuildHloModule(versioned_handle, + /*config=*/nullptr) + .ConsumeValueOrDie(); fprintf(stdout, "%s\n", module->ToString().c_str()); } |