- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 72 for BoolAttr (0.1 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/lite/quantization/tensorflow/tf_to_quant.cc
// attribute to create the quantization parameter for the new quantize op. rewriter.setInsertionPointAfter(tf_op.getOperation()); IntegerAttr num_bits = rewriter.getI64IntegerAttr(tf_op.getNumBits()); BoolAttr narrow_range = rewriter.getBoolAttr(tf_op.getNarrowRange()); Type res_type = tf_op.getType(); TypeAttr qtype = quant::GetQuantizedTypeAttr( rewriter, res_type, min_value, max_value, quant_dim, num_bits,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
failed(AddFloatAttr(func, attrs, "h_scale", &fbb)) || failed(AddFloatAttr(func, attrs, "w_scale", &fbb))) return failure(); auto use_regular_nms = mlir::dyn_cast_or_null<BoolAttr>(attrs.get("use_regular_nms")); if (!use_regular_nms) { return func.emitError() << "use_regular_nms attribute is not set or not a bool"; } fbb.Int("use_regular_nms", use_regular_nms.getValue());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/host_runtime/tfrt_ops.td
Other ops executed by TFRT may make use of $tensor_future. }]; let arguments = (ins Arg<TF_Tensor, [{The variable tensor to be loaded}], []>:$variable, DefaultValuedAttr<BoolAttr, "false">:$used_by_host ); let results = (outs TF_StrTensor:$array_key, TF_Tensor: $tensor_future ); TF_DerivedOperandTypeListAttr Tin = TF_DerivedOperandTypeListAttr<0>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 20:44:15 UTC 2024 - 6.8K bytes - Viewed (0) -
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)