Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TFRAttrType (0.5 sec)

  1. tensorflow/compiler/mlir/tfr/ir/tfr_types.h

      static std::string getTypeName() { return "TFRTensorListType"; }
    };
    
    class TFRAttrType : public Type::TypeBase<TFRAttrType, TFRType, TypeStorage> {
     public:
      using Base::Base;
      static constexpr StringLiteral name = "tfr.attr";
      static std::string getTypeName() { return "TFRAttrType"; }
    };
    
    }  // namespace TFR
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 05 07:17:01 UTC 2023
    - 4.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

        auto c_false =
            rewriter.create<arith::ConstantOp>(loc, rewriter.getBoolAttr(false));
        TypeAttr f32_attr = TypeAttr::get(rewriter.getF32Type());
        TFRAttrType output_type = TFRAttrType::get(rewriter.getContext());
        auto constant_f32_op = rewriter.create<ConstOp>(loc, output_type, f32_attr);
        TypeAttr i32_attr = TypeAttr::get(rewriter.getI32Type());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfr/passes/decompose.cc

            if (mlir::isa<TypeAttr>(attribute) || mlir::isa<ArrayAttr>(attribute) ||
                mlir::isa<StringAttr>(attribute) ||
                mlir::isa<FlatSymbolRefAttr>(attribute)) {
              TFRAttrType output_type = TFRAttrType::get(builder.getContext());
              attr_cst =
                  builder.create<ConstOp>(op->getLoc(), output_type, attribute);
            } else {
              attr_cst = builder.create<mlir::arith::ConstantOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

      let hasFolder = 1;
    
      let builders = [
        OpBuilder<(ins "Attribute":$value),
        [{
          auto* ctx = value.getContext();
          $_state.addAttribute("value", value);
          $_state.addTypes(TFRAttrType::get(ctx));
        }]>
      ];
    
      let assemblyFormat = [{
        $value attr-dict `->` type($out)
      }];
    }
    
    def TFR_ConstantTensorOp : TFR_Op<"constant_tensor", [Pure]> {
      let description = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
Back to top