- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for attrType (0.22 sec)
-
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) -
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)