diff options
Diffstat (limited to 'tensorflow/compiler/xla/legacy_flags/service_flags.h')
-rw-r--r-- | tensorflow/compiler/xla/legacy_flags/service_flags.h | 69 |
1 files changed, 69 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/legacy_flags/service_flags.h b/tensorflow/compiler/xla/legacy_flags/service_flags.h new file mode 100644 index 0000000000..d982506944 --- /dev/null +++ b/tensorflow/compiler/xla/legacy_flags/service_flags.h @@ -0,0 +1,69 @@ +/* Copyright 2017 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +#ifndef TENSORFLOW_COMPILER_XLA_LEGACY_FLAGS_SERVICE_FLAGS_H_ +#define TENSORFLOW_COMPILER_XLA_LEGACY_FLAGS_SERVICE_FLAGS_H_ + +// Legacy flags for XLA's service module. + +#include <vector> + +#include "tensorflow/compiler/xla/types.h" +#include "tensorflow/core/platform/types.h" +#include "tensorflow/core/util/command_line_flags.h" + +namespace xla { +namespace legacy_flags { + +// Append to *flag_list flag definitions associated with XLA's service module. +void AppendServiceFlags(std::vector<tensorflow::Flag>* flag_list); + +// The values of flags associated with XLA's service module. +typedef struct { + bool xla_hlo_profile; // Instrument the computation to collect per-HLO cycle + // counts + string xla_log_hlo_text; // If non-empty, print the text format of the HLO + // modules whose name partially + // matches this regex. E.g. xla_log_hlo_text=.* + // will dump the text for every module. + string xla_generate_hlo_graph; // If non-empty, dump graph of HLO modules + // whose name partially matches this regex. + // E.g. --xla_generate_hlo_graph=.* will dump + // the graph of every module. + bool xla_hlo_graph_addresses; // Show addresses of HLO ops in graph + bool xla_hlo_graph_layout; // Show layout of HLO ops in graph + bool xla_hlo_graph_for_compute_constant; // If true, include hlo dumps of + // graphs from ComputeConstant. + // Such graphs still need to be + // matched via + // xla_generate_hlo_graph. + string xla_dump_hlo_text_to; // Dumps HLO text for each HLO module that is + // executed into the provided directory path + string xla_dump_computations_to; // Dumps computations that XLA executes + // into the provided directory path + // Dumps parameters and results of computations that XLA executes into + // the provided directory path + string xla_dump_executions_to; +} ServiceFlags; + +// Return a pointer to the ServiceFlags struct; +// repeated calls return the same pointer. +// This should be called only after Flags::Parse() has returned. +ServiceFlags* GetServiceFlags(); + +} // namespace legacy_flags +} // namespace xla + +#endif // TENSORFLOW_COMPILER_XLA_LEGACY_FLAGS_SERVICE_FLAGS_H_ |