Home
last modified time | relevance | path

Searched refs:reduce_precision (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_verifier.cc388 Status ShapeVerifier::HandleReducePrecision(HloInstruction* reduce_precision) { in HandleReducePrecision() argument
389 return CheckShape(reduce_precision, ShapeInference::InferReducePrecisionShape( in HandleReducePrecision()
390 reduce_precision->operand(0)->shape(), in HandleReducePrecision()
391 reduce_precision->exponent_bits(), in HandleReducePrecision()
392 reduce_precision->mantissa_bits())); in HandleReducePrecision()
Dhlo_evaluator_typed_visitor.h2564 Status HandleReducePrecision(HloInstruction* reduce_precision) {
2566 parent_->evaluated_[reduce_precision],
2567 ElementWiseUnaryOp(reduce_precision, [&](ElementwiseT elem) {
2572 const uint32 dest_mantissa_bits = reduce_precision->mantissa_bits();
2573 const uint32 dest_exponent_bits = reduce_precision->exponent_bits();
2665 Status HandleReducePrecision(HloInstruction* reduce_precision) {
2666 return UnsupportedTypeError(reduce_precision);
2669 Status HandleReducePrecision(HloInstruction* reduce_precision) override {
2670 return HandleReducePrecision<ElementwiseT>(reduce_precision);
Dhlo_verifier.h67 Status HandleReducePrecision(HloInstruction* reduce_precision) override;
Dhlo_parser_test.cc1251 R"(HloModule reduce_precision in CreateTestCases()
/external/llvm-project/llvm/test/Transforms/InstCombine/
Dmaximum.ll391 define float @reduce_precision(float %x, float %y) {
392 ; CHECK-LABEL: @reduce_precision(
Dminimum.ll416 define float @reduce_precision(float %x, float %y) {
417 ; CHECK-LABEL: @reduce_precision(
Dmaxnum.ll379 define float @reduce_precision(float %x, float %y) {
380 ; CHECK-LABEL: @reduce_precision(
Dminnum.ll416 define float @reduce_precision(float %x, float %y) {
417 ; CHECK-LABEL: @reduce_precision(
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt1036 // CHECK-LABEL: func private @reduce_precision
1038 %reduce_precision (Arg_0.1: f32[3,4]) -> f32[3,4] {
1040 …// CHECK: "mhlo.reduce_precision"(%[[ARG0]]) {exponent_bits = 8 : i32, mantissa_bits = 10 : i32} :…
1041 …ROOT %reduce_precision = f32[3,4] reduce-precision(f32[3,4] %Arg_0.1), exponent_bits=8, mantissa_b…
Dexport.mlir1173 …%0 = "mhlo.reduce_precision"(%arg) {exponent_bits = 8 : i32, mantissa_bits = 10 : i32} : (tensor<3…
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1015 auto* reduce_precision = xla::Cast<xla::HloReducePrecisionInstruction>(instr); in EmitReducePrecisionOp() local
1017 builder_.getI32IntegerAttr(reduce_precision->exponent_bits())); in EmitReducePrecisionOp()
1019 builder_.getI32IntegerAttr(reduce_precision->mantissa_bits())); in EmitReducePrecisionOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dops.mlir472 // CHECK: lmhlo.reduce_precision
474 …%res = "mhlo.reduce_precision"(%value0) {exponent_bits=5 : i32, mantissa_bits=12 : i32}: (tensor<2…
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td531 def LHLO_ReducePrecisionOp: LHLO_Op<"reduce_precision", [SameTypeOperands]>,
Dhlo_ops.td1394 HLO_Op<"reduce_precision", [SameOperandsAndResultShape]>,
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_ops.mlir681 …"lmhlo.reduce_precision"(%arg0, %arg_out) { exponent_bits = 4 : i32, mantissa_bits = 4 : i32 } : (…
Dops.mlir1270 …%0 = "mhlo.reduce_precision"(%arg) {exponent_bits=2 : i32, mantissa_bits=3 : i32} : (tensor<2x4xf3…