diff options
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_opcode.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_opcode.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_opcode.cc b/tensorflow/compiler/xla/service/hlo_opcode.cc index 3888f757ad..314512d0a8 100644 --- a/tensorflow/compiler/xla/service/hlo_opcode.cc +++ b/tensorflow/compiler/xla/service/hlo_opcode.cc @@ -33,6 +33,8 @@ string HloOpcodeString(HloOpcode opcode) { return "add"; case HloOpcode::kBatchNormTraining: return "batch-norm-training"; + case HloOpcode::kBatchNormInference: + return "batch-norm-inference"; case HloOpcode::kBatchNormGrad: return "batch-norm-grad"; case HloOpcode::kBitcast: |