- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 153 for eraseOp (0.17 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/compose_uniform_quantized_type_pass.cc
// Erase unused ops in the reverse order. rewriter.eraseOp(output_uniform_dequantize_call_op); rewriter.eraseOp(output_uniform_quantize_call_op); rewriter.eraseOp(combined_scale_multiply_op); rewriter.eraseOp(subtract_op); rewriter.eraseOp(other_conv_op); rewriter.eraseOp(op); rewriter.eraseOp(input_i8_to_f32_convert_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc
rewriter.replaceAllOpUsesWith(output_mul, composite_op); // Note these must be erased in reverse topo order to avoid // failing in debug mode. rewriter.eraseOp(output_mul); rewriter.eraseOp(rhs_add); rewriter.eraseOp(op); rewriter.eraseOp(lhs_mul); rewriter.eraseOp(rhs_mul); return success(); } }; class OutlineCompositesPass
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/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc
.replaceAllUsesWith(new_call_op.getResult(0)); users_to_erase.push_back(user); } for (auto user : users_to_erase) rewriter.eraseOp(user); rewriter.eraseOp(call_op); func_op.eraseResult(0); func_op.insertResult(0, new_call_op.getResult(0).getType(), /*resultAttrs=*/nullptr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
id1.replaceAllUsesWith(id1.getInput()); min = tf_op.getMin(); rewriter.eraseOp(id1); } if (auto id2 = dyn_cast_or_null<TF::IdentityOp>(max.getDefiningOp())) { id2.replaceAllUsesWith(id2.getInput()); max = tf_op.getMax(); rewriter.eraseOp(id2); } if (!matchPattern(min, m_Constant(&min_value))) return failure();
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/tensorflow/passes/convert_custom_aggregation_op_to_quant_stats.cc
// When there are no min and max attributes, remove op. if (min == nullptr || max == nullptr) { op.getOutput().replaceAllUsesWith(op.getInput()); rewriter.eraseOp(op); return success(); } // The layer stats contain only the first min/max pairs. ElementsAttr layer_stats = DenseFPElementsAttr::get( RankedTensorType::get({2}, rewriter.getF32Type()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_functional_ops.cc
// TODO(jpienaar): Remove once recusive side-effects are supported. if (op.use_empty() && (op.getIsStateless() || (IsSideEffectFree(then_func) && IsSideEffectFree(else_func)))) { rewriter.eraseOp(op.getOperation()); return success(); } // Extract the constant cond value. DenseElementsAttr cond; if (!matchPattern(op.getCond(), m_Constant(&cond))) return failure();
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/lite/experimental/tac/transforms/device_transform.cc
LogicalResult matchAndRewrite(TFL::QuantizeOp quant_op, PatternRewriter& rewriter) const override { if (!quant_op.getResult().use_empty()) return failure(); rewriter.eraseOp(quant_op); return success(); } }; void OptimizeQuantizedOpToFloat(func::FuncOp func, MLIRContext* context) { RewritePatternSet patterns(func.getContext()); patterns
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
to_remove.push_back(init_func_op); } else { to_keep.push_back(sym_ref); } } for (auto func_op : to_remove) rewriter.eraseOp(func_op); if (to_keep.empty()) rewriter.eraseOp(op); else op->setAttr("initializers", rewriter.getArrayAttr(to_keep)); return success(); } }; void SessionInitializerOp::getCanonicalizationPatterns(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/post_quantize.cc
if ((custom_op_map.find(op_name) == custom_op_map.end()) || !custom_op_map.find(op_name)->second.no_side_effect) return failure(); } rewriter.eraseOp(op); return success(); } quant::CustomOpMap custom_op_map; }; #include "tensorflow/compiler/mlir/lite/transforms/generated_post_quantize.inc" void PostQuantizePass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
rewriter.replaceAllUsesWith(assign_group.getResult(0), const_group_size); rewriter.replaceAllUsesWith(assign_group.getResult(1), const_group_key); rewriter.eraseOp(assign_group); return success(); } }; void LegalizeTFCollective::runOnOperation() { // FIXME(b/226139061): Figure out a way to share the channel_id with
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0)