Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for FloatAttr (0.53 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        {
          epsilon =
              fused_batch_norm_op->getAttrOfType<::mlir::FloatAttr>("epsilon");
          if (!epsilon)
            epsilon = rewriter.getFloatAttr(rewriter.getF32Type(), 0.0001f);
    
          if (!(((mlir::isa<::mlir::FloatAttr>(epsilon))) &&
                ((mlir::cast<::mlir::FloatAttr>(epsilon).getType().isF32())))) {
            return rewriter.notifyMatchFailure(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/lower_tf.cc

    // Return an Attr representation of the value.
    static DenseElementsAttr GetF32Scalar(OpBuilder *builder, float value) {
      return DenseElementsAttr::get(
          tensorflow::GetTypeFromTFTensorShape({}, builder->getF32Type()),
          FloatAttr::get(builder->getF32Type(), value));
    }
    
    // Returns a TF_CastOp to F32. This function is used for CastOps that are
    // intermediate nodes in a TableGen pattern result. In such a case, the
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
Back to top