- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 72 for BoolAttr (0.21 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_patterns.td
// BatchNorm op patterns. //===----------------------------------------------------------------------===// def FalseBoolAttr : AttrConstraint<CPred<"!$_self.cast<BoolAttr>().getValue()">>; def TrueBoolAttr : AttrConstraint<CPred<"$_self.cast<BoolAttr>().getValue()">>; def CastValueToI64: NativeCodeCall< "CastValueToI64($0.getLoc(), $1, &$_builder)">; def CastValueToElementType: NativeCodeCall<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 34.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
mlir::DenseIntElementsAttr::get(static_shape_type, static_shape); return rewriter.create<TF::ConstOp>(loc, static_shape_attr).getOutput(); } // If the shape is not static, create a new ShapeOp. BoolAttr false_attr = rewriter.getBoolAttr(false); return rewriter .create<TF::ShapeOp>(loc, input, /*use_32bit=*/false_attr) .getOutput(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
using OptionalOpShardingVector = llvm::SmallVector<std::optional<OpShardingVariant>, 8>; using llvm::StringRef; using mlir::Block; using mlir::BlockArgument; using mlir::BoolAttr; using mlir::Builder; using mlir::IntegerAttr; using mlir::LogicalResult; using mlir::ModuleOp; using mlir::Operation; using mlir::OpOperand; using mlir::OpResult; using mlir::RankedTensorType;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize.td
//===----------------------------------------------------------------------===// def ReadVariableOfCast : Pat< (TF_ReadVariableOp:$src (TF_CastOp:$output $x, BoolAttr:$Truncate)), (TF_ReadVariableOp:$dest $x), [(HasOnlyReadVariableOpUsers $output)], [(CopyAttrs $src, $dest)]>; //===----------------------------------------------------------------------===//
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:42:28 UTC 2023 - 17K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.td
The `tf.IfrtCall` kernel uses the output $array_key. Other ops executed by TFRT may make use of $tensor_future. }]; let arguments = (ins TFTensorType:$variable, DefaultValuedAttr<BoolAttr, "false">:$used_by_host ); let results = (outs TFTensorType:$array_key, MlrtFutureType: $tensor_future ); } def IfrtRestoreVariableOp: TensorflowMlrt_Op<"ifrt_restore_variable", []> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
auto operands = adaptor.getOperands(); assert(operands.size() == 2 && "equal op has two operands"); auto ctx = getContext(); if (operands[0] == operands[1]) return BoolAttr::get(ctx, true); return BoolAttr::get(ctx, false); } OpFoldResult ConstOp::fold(FoldAdaptor adaptor) { auto operands = adaptor.getOperands(); (void)operands; assert(operands.empty() && "constant has no operands");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
lhs_dot_dimensions_info.FlattenedOutDimensionSize()}, llvm::ArrayRef<int64_t>{ rhs_dot_dimensions_info.FlattenedOutDimensionSize()}); BoolAttr false_attr = rewriter.getBoolAttr(false); auto matmul = rewriter.create<TFL::BatchMatMulOp>( loc, RankedTensorType::get(matmul_shape, result_type.getElementType()),
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/lite/stablehlo/transforms/legalize_hlo.cc
ImplicitLocOpBuilder& builder, bool is_lhs) { auto operand_type = mlir::cast<ShapedType>(operand.getType()); BoolAttr true_attr = builder.getBoolAttr(true); auto operand_shape = builder.create<TF::ShapeOp>(operand, true_attr); const int64_t operand_rank = operand_type.getRank();
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/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
<< "]"; return success(); } bool VariableIsInitialized(TF::VarHandleOp var_handle_op) { auto is_variable_initialized = var_handle_op->getAttrOfType<BoolAttr>("_is_initialized"); // Assume variable is initialized if attribute is not set. // There are paths that doesn't mark the variables. All variables // that doesn't have the attribute will be promoted.
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/transforms/lower_tf.td
NativeCodeCall<"GetI64ElementsAttr({" # value # "}, &$_builder)">; class GetF32Scalar<int value> : NativeCodeCall<"GetF32Scalar(&$_builder, " # value # ")">; def TrueBoolAttr : AttrConstraint<CPred<"$_self.cast<::mlir::BoolAttr>().getValue()">>; def CreateTFShapeOp : NativeCodeCall< "$_builder.create<TF::ShapeOp>($0.getLoc(), $1, $2)">; def IsI32 : NativeCodeCall<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 24.7K bytes - Viewed (0)