- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 68 for DenseIntElementsAttr (0.24 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_util.h
float_attr.getSplatValue<APFloat>().isExactlyValue(raw_value)) return true; } else if (mlir::isa<IntegerType>(element_type)) { DenseIntElementsAttr int_attr; if (matchPattern(value, m_Constant(&int_attr)) && int_attr.isSplat() && int_attr.getSplatValue<APInt>() == raw_value) return true; } return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
auto contracting_segids_cst = builder.create<TFL::ConstOp>( builder.getI32TensorAttr(flattened_contracting_segids)); auto num_segids_tensor = builder.create<TFL::ConstOp>(DenseIntElementsAttr::get( RankedTensorType::get({}, builder.getIntegerType(32)), 1)); auto flattened_out_dims = builder.create<TFL::UnsortedSegmentProdOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils_test.cc
Value value = CreateConstValue<int8_t>(builder, module->getLoc(), unpacked_shape, unpacked_values); Value packed_value = PackOperand(builder, module->getLoc(), value, pack_dim); DenseIntElementsAttr packed_value_attr; ASSERT_TRUE(matchPattern(packed_value, m_Constant(&packed_value_attr))); ShapedType packed_shape_type = mlir::dyn_cast<ShapedType>(packed_value.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
Operation* def_op = op.getInput().getDefiningOp(); auto qconst_op = llvm::dyn_cast_or_null<QConstOp>(def_op); if (qconst_op == nullptr) return failure(); DenseIntElementsAttr perm_tensor; if (!matchPattern(op.getPerm(), m_Constant(&perm_tensor))) return failure(); if (!mlir::isa<quant::UniformQuantizedType>( (getElementTypeOrSelf(op.getOutput().getType()))))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_arith_ops_folder.cc
auto input_type = mlir::dyn_cast<RankedTensorType>(input.getType()); if (!input_type) return success(); int64_t rank = input_type.getRank(); DenseIntElementsAttr dims_attr; if (!matchPattern(dims, m_Constant(&dims_attr))) return success(); for (const auto &dim_pair : llvm::enumerate(dims_attr)) { int64_t cur_dim = dim_pair.value().getSExtValue();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
const_op->getLoc(), DenseElementsAttr::get( RankedTensorType::get(/*shape=*/{}, element_type), splat_elements_attr.getSplatValue<Attribute>())); auto broadcast_dims = DenseIntElementsAttr::get( RankedTensorType::get(/*shape=*/{0}, op_builder->getI64Type()), llvm::SmallVector<int64_t>{}); mhlo::BroadcastInDimOp broadcast_in_dim_op = op_builder->create<mhlo::BroadcastInDimOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td
class I32VectorElementsAttr<int len> : ElementsAttrBase< CPred<"$_self.isa<DenseIntElementsAttr>() &&" "$_self.cast<DenseIntElementsAttr>().getType()." "getElementType().isSignlessInteger(32)">, "32-bit int elements attribute of shape [" # len # "]"> { let storageType = [{ DenseIntElementsAttr }]; let returnType = [{ DenseIntElementsAttr }]; let constBuilderCall = "DenseElementsAttr::get("
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 28.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
const int ceil_pad_right = pool.kw - remaining_right; pad_right = ceil_pad_right - pool.pw; } return {pool.ph, pad_bottom, pool.pw, pad_right}; } DenseIntElementsAttr GetPadOpAttr(Builder& builder, CompositeOp op) { const TorchAvgPoolData pool = GetTorchAvgPoolData(op); const auto values = GetPadOpPaddingValues(pool);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h
const SmallVector<T>& values) { if constexpr (std::is_integral_v<T>) { auto shape_type = RankedTensorType::get(shape, builder.getIntegerType(sizeof(T) * 8)); const auto attr = DenseIntElementsAttr::get(shape_type, values); return builder.create<TF::ConstOp>(loc, attr); } const auto type = RankedTensorType::get(shape, builder.getF32Type());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/unfold_large_splat_constant.cc
return; } op_builder->setInsertionPoint(const_op); mlir::arith::ConstantOp fill_shape = op_builder->create<mlir::arith::ConstantOp>( const_op->getLoc(), DenseIntElementsAttr::get( tensorflow::GetTypeFromTFTensorShape( {splat_elements_attr.getType().getRank()}, op_builder->getI64Type()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.5K bytes - Viewed (0)