Lines Matching defs:epsilon
47 // cudnn defines CUDNN_BN_MIN_EPSILON = 1e-5 as the minimum acceptable epsilon
50 return batch_norm->epsilon() >= 1e-5;
69 HloInstruction* epsilon = computation_->AddInstruction(
70 HloInstruction::CreateConstant(Literal::CreateR0(batch_norm->epsilon())));
77 operands.push_back(epsilon);
104 HloInstruction* epsilon = computation_->AddInstruction(
105 HloInstruction::CreateConstant(Literal::CreateR0(batch_norm->epsilon())));
112 operands.push_back(epsilon);
121 // {output, mean, rsqrt(variance + epsilon)},
134 variance_plus_epsilon, epsilon));
167 HloInstruction* epsilon = computation_->AddInstruction(
168 HloInstruction::CreateConstant(Literal::CreateR0(batch_norm->epsilon())));
173 // The cudnn libcall expects its input to be rsqrt(variance + epsilon), but
178 batch_norm->mutable_operand(3), epsilon));
188 operands.push_back(epsilon);
