- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 67 for BoolAttr (0.11 sec)
-
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.td
"getElementTypeOrSelf($_self.cast<ElementsAttr>().getType()).isa<FloatType>()">, "float constant tensor">; // Checks if the boolean value is false. def IsFalseBoolAttr : AttrConstraint< CPred<"!$_self.cast<BoolAttr>().getValue()">>; // Checks if the value has only one user. def HasOneUse : Constraint<CPred<"$0.hasOneUse()">>; // Gets the type of a value. def GetValueType : NativeCodeCall<"$0.getType()">;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 04:55:44 UTC 2024 - 6.7K 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/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
getOperation().walk([&](IslandOp island_op) { auto parent_func = island_op->getParentOfType<func::FuncOp>(); auto skip_island_outlining = parent_func->getAttrOfType<BoolAttr>(mlir::TF::kSkipIslandOutlining); if (skip_island_outlining && skip_island_outlining.getValue()) { // Island was marked to be skipped. return WalkResult::advance(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K 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/lite/transforms/optimize_batch_matmul.cc
/*fused_activation_function=*/rewriter.getStringAttr("NONE"), /*weights_format=*/rewriter.getStringAttr("DEFAULT"), /*keep_num_dims=*/rewriter.getBoolAttr(true), /*asymmetric_quantize_inputs=*/mlir::BoolAttr()); rewriter.replaceOp(bmm_op, {fc_op.getResult(0)}); return success(); }; }; // Converts batch_matmul operation with a ones tensor to a reduce_sum. struct ConvertBatchMatMulOpToReduceSum
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
LogicalResult PartitionResourceReadsWrites( tf_device::ClusterFuncOp cluster_func) { bool use_spmd = false; if (auto use_spmd_attr = cluster_func->getAttrOfType<BoolAttr>(kUseSpmdAttr)) use_spmd = use_spmd_attr.getValue(); if (!use_spmd) return success(); auto num_cores_per_replica_attr = cluster_func->getAttrOfType<IntegerAttr>(kNumCoresPerReplicaAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K 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/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)