- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 73 for BoolAttr (0.11 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/python/mlir_wrapper/ops.cc
.def("create", [](mlir::OpBuilder& opb, mlir::Location loc, mlir::Value x, mlir::Value y) { return opb .create<mlir::TF::NotEqualOp>( loc, x, y, mlir::BoolAttr::get(opb.getContext(), true)) .getOperation(); }); // mlir::TF::SubOp py::class_<mlir::TF::SubOp>(m, "Tf_SubOp") .def("create",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 30 02:12:49 UTC 2022 - 7.6K 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)