- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 30 for BoolAttr (0.36 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
} else if (auto int_attr = mlir::dyn_cast_or_null<IntegerAttr>(attr.second)) { fbb.Int(attr.first.data(), int_attr.getInt()); } else if (auto bool_attr = mlir::dyn_cast_or_null<BoolAttr>(attr.second)) { fbb.Bool(attr.first.data(), bool_attr.getValue()); } else if (auto string_attr = mlir::dyn_cast_or_null<StringAttr>(attr.second)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
// This pass extracts a CPU computation cluster with `_xla_outside_compilation` // annotation from the head or tail of a TPU cluster. namespace { using mlir::Block; using mlir::BlockArgument; using mlir::BoolAttr; using mlir::ModuleOp; using mlir::OpBuilder; using mlir::Operation; using mlir::OperationPass; using mlir::Region; using mlir::StringAttr; using mlir::Type; using mlir::Value; using mlir::WalkResult;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
#include "tensorflow/core/lib/monitoring/gauge.h" namespace tensorflow { namespace tf2xla { namespace internal { namespace { using mlir::Block; using mlir::BoolAttr; using mlir::Dialect; using mlir::LogicalResult; using mlir::MLIRContext; using mlir::ModuleOp; using mlir::Operation; using mlir::OperationName; using mlir::OperationPass; using mlir::Pattern;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
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/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/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/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)