- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 40 for hasOneUse (0.18 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
// if there is only one. Operation* GetUserIfOnlyOne(Operation* op) { if (op->getNumResults() != 1) return nullptr; auto result = op->getResult(0); if (!result.hasOneUse()) return nullptr; return (*result.getUses().begin()).getOwner(); } // Gets operation providing value for the given operand of given operation // if the given operation is the only user.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_helper.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 28K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_op_order.cc
: OpRewritePattern<DequantizeOp>(context) {} LogicalResult matchAndRewrite(DequantizeOp dequantize_op, PatternRewriter& rewriter) const override { if (!dequantize_op->hasOneUse()) return failure(); auto use = dequantize_op->use_begin(); Operation* passthrough_op = use->getOwner(); unsigned operand_index = use->getOperandNumber();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
void QuantizationDriver::RequantizeArg(const BlockArgument arg, RequantizeStates& states) { Value value = arg; builder_.setInsertionPointToStart(arg.getOwner()); if (value.hasOneUse()) { Operation* user = value.use_begin().getUser(); if (auto q = dyn_cast<quantfork::QuantizeCastOp>(user)) { value = q.getResult(); builder_.setInsertionPoint(arg.getOwner(), ++Block::iterator(user));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_read_for_write.cc
ResourceValueAndSubtype GetResourceWriteResult( tf_device::ClusterFuncOp cluster_func, Value result) { ResourceValueAndSubtype resource; if (!result.hasOneUse()) return resource; Operation* result_user = *result.getUsers().begin(); auto assign_var = dyn_cast<TF::AssignVariableOp>(result_user); if (!assign_var) return resource; auto handle = assign_var.getResource();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
PatternRewriter &rewriter) const override { // We don't want to insert quantize/dequantize if the quantize op exists. auto res = tf_op.getOutputs(); if (!res.hasOneUse() || isa<quantfork::QuantizeCastOp>(*res.user_begin())) return failure(); // Extract the min/max constant values from the operands. We also consider
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/quantization/stablehlo/passes/quantization_patterns.cc
// If activation exists, omit clipping op. // Since out_scale and out_zp are computed based on clipped range, // explicit activation clipping op is not required. if (isa<AddOp>(next_op) && gemm_style_op->hasOneUse()) { // bias fusion CreateAndReturnQuantizedBiasPattern( next_op, rewriter, entry_func_op, func_result_type, accumulation_quantized_element_type, gemm_style_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/fake_quant_utils.h
OpBuilder &rewriter) const { // We don't want to insert quantize/dequantize if the quantize op exists. auto res = tf_op.getOutputs(); if (!res.hasOneUse() || isa<QuantizeOp>(*res.user_begin())) { return failure(); } // Extract the min/max constant values from the operands. We also consider
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
constant_op.getResult().hasOneUse()) continue; // Do not duplicate constant op if the size is too large. // 32 is chosen to be larger than all constants useful for shape references, // while not too large to possibly significantly increase model size. if (constant_op.getValue().getNumElements() > 32) continue; while (!constant_op.getResult().hasOneUse()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
}; bool PrepareQuantizePass::SetInputNodesQuantizationParams(func::FuncOp func) { StringRef func_name = func.getName(); auto has_quantize_op = [&](const Value arg) { return (arg.hasOneUse() && llvm::isa<quantfork::QuantizeCastOp>(*arg.user_begin())); }; bool need_to_set_input_nodes_quantization_params = false; for (const BlockArgument arg : func.getArguments()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0)