diff options
author | 2018-06-06 11:42:25 -0700 | |
---|---|---|
committer | 2018-06-06 11:45:19 -0700 | |
commit | bbe49e75336ea2206a146a4d03614aaeca013079 (patch) | |
tree | 0704b69d08e7e74afa4a6b873ed8a0c2340980e5 /tensorflow/compiler/xla/service/hlo_casting_utils.h | |
parent | 88ac13ac825f5eecb7082d5878605251a66b3012 (diff) |
Split out HloBatchNormInstruction as subclasses from HloInstruction.
PiperOrigin-RevId: 199500687
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_casting_utils.h')
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_casting_utils.h | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_casting_utils.h b/tensorflow/compiler/xla/service/hlo_casting_utils.h index b15f1f24c6..7f73bba036 100644 --- a/tensorflow/compiler/xla/service/hlo_casting_utils.h +++ b/tensorflow/compiler/xla/service/hlo_casting_utils.h @@ -18,10 +18,13 @@ limitations under the License. #ifndef TENSORFLOW_COMPILER_XLA_SERVICE_HLO_CASTING_UTILS_H_ #define TENSORFLOW_COMPILER_XLA_SERVICE_HLO_CASTING_UTILS_H_ -#include "tensorflow/compiler/xla/service/hlo_instruction.h" +#include <type_traits> +#include "tensorflow/core/platform/logging.h" namespace xla { +class HloInstruction; + template <class T> using EnableIfDerivedFromHlo = typename std::enable_if<std::is_base_of<HloInstruction, T>::value>::type; |