- Sort Score
- Result 10 results
- Languages All
Results 11 - 18 of 18 for BoolAttr (0.11 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/utils/fake_quant_utils.h
// 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, input_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 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_ops.td
This op returns a scalar string tensor as a key for user to look for the loaded array and a future containing the restored tensor. }]; let arguments = (ins TF_Tensor:$variable, DefaultValuedAttr<BoolAttr, "false">:$used_by_host ); let results = (outs TF_Tensor:$array_key, MlrtFutureType: $tensor_future ); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:35:32 UTC 2024 - 6.7K 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/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/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)