aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/legacy_flags
diff options
context:
space:
mode:
authorGravatar Nick Desaulniers <ndesaulniers@google.com>2018-01-19 15:01:35 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-01-19 15:05:11 -0800
commita141651809e17be5ebe02715074357a7c52c0220 (patch)
treefd0046b2b5b289fb8162ce917f40a7ef3d064fda /tensorflow/compiler/xla/legacy_flags
parent0c4c353690b0cec2e26e10abd59f66f7e61cc974 (diff)
[XLA] Add xla_dump_prepass_hlo_proto_to debug flag
Add xla_dump_prepass_hlo_proto_to debug flag for dumping HLO protos *BEFORE* running HLO passes, which will perform backend specific optimizations. PiperOrigin-RevId: 182594324
Diffstat (limited to 'tensorflow/compiler/xla/legacy_flags')
-rw-r--r--tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc5
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc b/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc
index e88bffd0ba..fe3a4d2f6d 100644
--- a/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc
+++ b/tensorflow/compiler/xla/legacy_flags/debug_options_flags.cc
@@ -224,6 +224,11 @@ void AllocateFlags() {
"xla_dump_hlo_proto_to", flag_values->mutable_xla_dump_hlo_proto_to(),
"Dump compilation artifacts as proto binary into this directory."),
tensorflow::Flag(
+ "xla_dump_prepass_hlo_proto_to",
+ flag_values->mutable_xla_dump_prepass_hlo_proto_to(),
+ "Dump compilation artifacts, before hlo passes are executed, as "
+ "proto binary into this directory."),
+ tensorflow::Flag(
"xla_test_all_output_layouts",
bool_setter_for(&DebugOptions::set_xla_test_all_output_layouts),
flag_values->xla_test_all_output_layouts(),