14 #ifndef TENSORFLOW_COMPILER_XLA_SERVICE_HLO_PASS_FIX_H_ 15 #define TENSORFLOW_COMPILER_XLA_SERVICE_HLO_PASS_FIX_H_ 17 #include "tensorflow/compiler/xla/status_macros.h" 18 #include "tensorflow/compiler/xla/statusor.h" 19 #include "tensorflow/compiler/xla/types.h" 20 #include "tensorflow/core/platform/macros.h" 26 template <
typename Pass>
29 template <
typename... Args>
30 explicit HloPassFix(Args&&... args) : Pass(args...) {}
31 StatusOr<bool> Run(
HloModule* module)
override {
33 bool changed_this_iteration =
true;
34 while (changed_this_iteration) {
35 TF_ASSIGN_OR_RETURN(changed_this_iteration, Pass::Run(module));
36 changed |= changed_this_iteration;
42 #endif // TENSORFLOW_COMPILER_XLA_SERVICE_HLO_PASS_FIX_H_ namespace for xla
Definition: client_library.cc:26
Definition: hlo_pass_fix.h:27
Definition: hlo_module.h:52