Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for attrType (0.22 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

      ArrayRef<int64_t> shape = mlir::cast<ShapedType>(value.getType()).getShape();
      auto attr_type = RankedTensorType::get({static_cast<int64_t>(shape.size())},
                                             rewriter.getIntegerType(64));
      auto attr = DenseElementsAttr::get(attr_type, shape);
      return rewriter.create<arith::ConstantOp>(value.getLoc(), attr_type, attr);
    }
    
    bool IsSign(APInt a, APInt sign) {
      if (a.isZero()) return a == sign;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

      if (!status->status.ok()) return metadata;
      switch (attr->value_case()) {
    #define SINGLE_CASE(kK, attr_type, size_expr) \
      case tensorflow::AttrValue::kK:             \
        metadata.is_list = 0;                     \
        metadata.list_size = -1;                  \
        metadata.type = attr_type;                \
        metadata.total_size = size_expr;          \
        break;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top