1 /* Copyright 2019 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 #ifndef TENSORFLOW_COMPILER_JIT_INTRODUCE_FLOATING_POINT_JITTER_PASS_H_ 16 #define TENSORFLOW_COMPILER_JIT_INTRODUCE_FLOATING_POINT_JITTER_PASS_H_ 17 18 #include "tensorflow/core/common_runtime/optimization_registry.h" 19 20 namespace tensorflow { 21 // A debug-only pass that introduces error into outputs of specific TF nodes. 22 // This can be used to check the sensitivity of a TF graph to floating point 23 // rounding differences. 24 // 25 // This pass is controlled by TF_XLA_FLAGS. Please see 26 // IntroduceFloatingPointJitterPassFlags for information on how to use this. 27 class IntroduceFloatingPointJitterPass : public GraphOptimizationPass { 28 public: 29 IntroduceFloatingPointJitterPass() = default; 30 31 Status Run(const GraphOptimizationPassOptions& options) override; 32 }; 33 } // namespace tensorflow 34 35 #endif // TENSORFLOW_COMPILER_JIT_INTRODUCE_FLOATING_POINT_JITTER_PASS_H_ 36