- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 23 for SetArg (0.24 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
if (!dq_op) { auto mul_op = builder.create<TF::MulOp>(loc, value, multiplier); return mul_op.getResult(); } auto q_op = dq_op.getArg().getDefiningOp<quantfork::QuantizeCastOp>(); if (!q_op) return {}; Value float_value = q_op.getArg(); Value new_value = builder.create<TF::MulOp>(loc, float_value, multiplier); auto new_value_type = mlir::cast<TensorType>(new_value.getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc
// can backtracked to. Such results will be called "function passthrough". If // the result cannot be backtracked to a region argument, returns // std::nullopt. std::optional<int> GetArg(int result_index) const { if (auto arg = mlir::dyn_cast<BlockArgument>(GetValue(result_index))) if (arg.getParentBlock() == ®ion_->front()) return arg.getArgNumber(); return std::nullopt; } private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions.cc
preceding_op)) { if (auto q_op = llvm::dyn_cast_or_null<quantfork::QuantizeCastOp>( dq_op.getArg().getDefiningOp())) { Operation* q_op_input = q_op.getArg().getDefiningOp(); is_weight_constant = q_op_input && q_op_input->hasTrait<OpTrait::ConstantLike>(); } } if (!is_weight_constant) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 16.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.cc
// quant.qcast op in case it overrides the information from training FakeQuant // ops. func.walk([&](quantfork::QuantizeCastOp q) { auto input_op = q.getArg().getDefiningOp(); if (auto stats = dyn_cast_or_null<quantfork::StatisticsOp>(input_op)) { q.setOperand(stats.getArg()); if (stats.use_empty()) stats.erase(); } }); // Step 1: forward pass: propagate any value scales which are not produces
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 43.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.cc
PatternRewriter& rewriter) const override { auto preceding_qcast = q_op.getArg().getDefiningOp<quantfork::QuantizeCastOp>(); if (!preceding_qcast) return failure(); auto new_qcast = rewriter.create<quantfork::QuantizeCastOp>( q_op.getLoc(), q_op.getType(), preceding_qcast.getArg()); new_qcast->setAttr(kVolatileOpAttrName, rewriter.getUnitAttr()); q_op->replaceAllUsesWith(new_qcast);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
return failure(); } output_types = {new_output_type}; } else { output_types = {output_type.clone(elem_type.getStorageType())}; } SmallVector<Value> args = {q_op.getArg(), scale, zero_point}; FlatSymbolRefAttr func_name = FlatSymbolRefAttr::get(rewriter.getStringAttr(kQuantizeFuncName)); auto quantize_call = rewriter.create<TF::PartitionedCallOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
void PrepareQuantizeDRQPass::removeAllStatsOp(func::FuncOp func) { func.walk([&](quantfork::StatisticsOp stats_op) { stats_op.replaceAllUsesWith(stats_op.getArg()); stats_op.erase(); }); } #include "tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize.inc" void PrepareQuantizeDRQPass::runOnOperation() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
auto q = rewriter.create<QuantizeOpT>(op.getLoc(), result_type, op.getArg()); q->setAttr(kVolatileOpAttrName, rewriter.getUnitAttr()); auto dq = rewriter.create<DequantizeOpT>(op.getLoc(), op.getType(), q); op.getResult().replaceAllUsesWith(dq); q.getOperation()->replaceUsesOfWith(dq, op.getArg()); op.erase(); return success(); } private: int num_bits;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.td
let extraClassDeclaration = [{ // Return element type of the input tensor type. Only available when the // input is a MLIR built-in tensor type. Attribute getInputElementType() { if (auto ty = getArg().getType().dyn_cast<TensorType>()) { return TypeAttr::get(ty.getElementType()); } return {}; } }]; let hasCanonicalizer = 1; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 10:54:29 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver.cc
for (int i = 0; i < op->getNumOperands(); ++i) { if (auto dq_op = dyn_cast_or_null<quantfork::DequantizeCastOp>( op->getOperand(i).getDefiningOp())) { auto type = mlir::cast<TensorType>(dq_op.getArg().getType()).getElementType(); if (auto per_axis_qtype = mlir::dyn_cast_or_null<quant::UniformQuantizedPerAxisType>( QuantizedType::getQuantizedElementType(type))) { return true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 38.1K bytes - Viewed (0)