Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TensorIsSign (0.18 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_patterns.td

    def ValueGreaterThanZero : Constraint<CPred<"ValueGreaterThanZero($0)">>;
    def ValueIsReciprocal : Constraint<CPred<"ValueIsReciprocal($0, $1)">>;
    def TensorIsSign : Constraint<CPred<"TensorIsSign($_builder, $0, $1)">>;
    def SameValue : Constraint<CPred<"$0 == $1">>;
    def FloatOrDefaultCompare : Constraint<CPred<
      "!$0 || $0.getValue() == ::mlir::mhlo::ComparisonType::FLOAT">>;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Feb 03 08:58:22 UTC 2024
    - 34K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        return !value.isNaN() && !value.isNegative() && !value.isZero();
      }
      return false;
    }
    
    // Returns whether the splat constant is the sign of the int or float Tensor.
    bool TensorIsSign(PatternRewriter& rewriter, ElementsAttr float_or_int,
                      ElementsAttr sgn_cst) {
      auto sgn_splat = llvm::dyn_cast<SplatElementsAttr>(sgn_cst);
      if (!sgn_splat) return false;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
Back to top