[XLA] Fix comments on HloInstruction::epsilon() and HloInstruction::feature_index().
These functions can be called for kBatchNorm{Training,Inference,Grad},
not just kBatchNormTraining.
No functional change.
PiperOrigin-RevId: 179363059
Loading
Please sign in to comment