- Sort Score
- Result 10 results
- Languages All
Results 21 - 26 of 26 for DenseIntElementsAttr (1.18 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
} // Generate a dummy constant island of requested type. IslandOp GetDummyConstant(OpBuilder builder, ShapedType const_type, Location loc) { DenseIntElementsAttr val = DenseIntElementsAttr::get(const_type, 1); auto const_op = builder.create<TF::ConstOp>(loc, val); auto const_island = CreateIsland(const_op, {}, builder); return const_island; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
Operation *tposed_const; if (other_input_type.getNumElements() == 1) { tposed_const = cst_arg; } else { auto inverse_perm_attr = DenseIntElementsAttr::get( RankedTensorType::get(inverse_perm.size(), rewriter.getI32Type()), inverse_perm); auto inverse_perm_op = rewriter.create<arith::ConstantOp>(perm.getLoc(), inverse_perm_attr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
namespace mlir { namespace TF { namespace collection_ops_util { Value CreateScalarConst(int32_t value, OpBuilder builder, Location loc) { auto attr = DenseIntElementsAttr::get( tensorflow::GetTypeFromTFTensorShape({}, builder.getI32Type()), value); return builder.create<TF::ConstOp>(loc, attr); } Value GetR1Const(ArrayRef<int64_t> r1, OpBuilder builder, Location loc,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
return transpose_op.emitOpError( "perm tensor elements size is not equal to input tensor rank"); } } // Get permutation axes of the TransposeOp DenseIntElementsAttr perm; if (!matchPattern(transpose_op.getPerm(), m_Constant(&perm))) { return failure(); } SmallVector<int64_t, 4> axes; for (const auto& axis_int : perm.getValues<APInt>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
mlir::TensorType ty = mlir::RankedTensorType::get(vals.size(), builder.getIntegerType(64)); TensorProto tensor; TF_RETURN_IF_ERROR(ConvertToTensorProto( mlir::DenseIntElementsAttr::get(ty, vals), &tensor)); *list->add_tensor() = tensor; } else { return errors::Unimplemented("Unhandled attribute!"); } } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
OpBuilder* builder) { auto type = mlir::RankedTensorType::get(value.size(), builder->getIntegerType(32)); auto dense_values = mlir::DenseIntElementsAttr::get(type, value); return builder->create<TF::ConstOp>(loc, dense_values); } TF::ConstOp CreateScalarConstantOp(int value, Location loc, OpBuilder* builder) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0)