Searched refs:reduce_precision (Results 1 – 4 of 4) sorted by relevance
213 Status ShapeVerifier::HandleReducePrecision(HloInstruction* reduce_precision) { in HandleReducePrecision() argument214 return CheckShape(reduce_precision, ShapeInference::InferReducePrecisionShape( in HandleReducePrecision()215 reduce_precision->operand(0)->shape(), in HandleReducePrecision()216 reduce_precision->exponent_bits(), in HandleReducePrecision()217 reduce_precision->mantissa_bits())); in HandleReducePrecision()
2531 Status HandleReducePrecision(HloInstruction* reduce_precision) { in HandleReducePrecision() argument2533 parent_->evaluated_[reduce_precision], in HandleReducePrecision()2534 ElementWiseUnaryOp(reduce_precision, [reduce_precision]( in HandleReducePrecision()2537 const uint32_t mantissa_bits = reduce_precision->mantissa_bits(); in HandleReducePrecision()2538 const uint32_t exponent_bits = reduce_precision->exponent_bits(); in HandleReducePrecision()2622 Status HandleReducePrecision(HloInstruction* reduce_precision) { in HandleReducePrecision() argument2630 Status HandleReducePrecision(HloInstruction* reduce_precision) { in HandleReducePrecision() argument2631 return UnsupportedTypeError(reduce_precision); in HandleReducePrecision()2634 Status HandleReducePrecision(HloInstruction* reduce_precision) override { in HandleReducePrecision() argument2635 return HandleReducePrecision<ElementwiseT>(reduce_precision); in HandleReducePrecision()
61 Status HandleReducePrecision(HloInstruction* reduce_precision) override;
1092 R"(HloModule reduce_precision in CreateTestCases()