- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 78 for constop (0.12 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
// shared_name mapping. The shared_name is the shared name of the corresponding // VarHandleOp. llvm::MapVector<TF::ConstOp, std::string> ReplaceConstOpUsesWithVariableReads( llvm::ArrayRef<TF::ConstOp> target_const_ops) { llvm::MapVector<TF::ConstOp, std::string> const_op_name_map{}; // Keeps track of the number of occurrences of each synthesized name. The
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc
ShapedType new_type = shaped_quantized_type.clone(storage_type); rewriter.setInsertionPointAfter(op); auto const_op = rewriter.create<TF::ConstOp>(op.getLoc(), new_type, tensor_proto_attr); auto new_identity_op = rewriter.create<TF::IdentityOp>( op->getLoc(), const_op.getType(), const_op); return new_identity_op.getResult(); } Operation* LogicsForUniformDequanization(PatternRewriter& rewriter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc
class QuantizeConstWeights : public OpRewritePattern<TF::ConstOp> { public: explicit QuantizeConstWeights( MLIRContext* context, const tensorflow::quantization::QuantizationOptions& quantization_options) : OpRewritePattern<TF::ConstOp>(context), quant_options_(quantization_options) {} LogicalResult matchAndRewrite(TF::ConstOp op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
PatternRewriter &rewriter) const override { auto cast_op = dyn_cast<TFR::CastOp>(qparams_op.getInput().getDefiningOp()); auto cast_qtype = getQuantizedElementType(cast_op); if (!cast_qtype) { return failure(); } TF::ConstOp scale_op; TF::ConstOp zp_op; // Reads quantization parameters from the quantized type, and converts
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
Value CreateZeroInput(Location loc, OpBuilder& builder, Attribute zero_attr, DenseIntElementsAttr shape_attr) { ConstOp zero = builder.create<ConstOp>(loc, zero_attr); zero->setAttr(kICIWeightDistributionMlirBridgeMarker, builder.getBoolAttr(true)); ConstOp shape = builder.create<ConstOp>(loc, shape_attr); shape->setAttr(kICIWeightDistributionMlirBridgeMarker, builder.getBoolAttr(true));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
// Helper method that returns an op from 'transpose_ops' that match criteria // for an 'operand' and 'permutation' TransposeOp ReuseExistingTranspose(const OpOperand* operand, const SmallVector<int64_t, 4>& permutation, Operation* op, ConstOp permutation_op, SmallVector<TransposeOp, 2>* transpose_ops) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
// Consider UQ op legal if it has a CastOp next to the qint input/output. if (IsTFUniformQuantizedOp(op)) { return IsTFUniformQuantizedOpLegal(op); } else if (auto cast_op = llvm::dyn_cast<TF::CastOp>(op)) { return IsCastOpLegal(cast_op); } else if (auto const_op = llvm::dyn_cast<TF::ConstOp>(op)) { return !IsIllegalType(const_op.getOutput().getType()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
float ratio_threshold = kBlockOverRandomSparsityRatio; if (isa<ConstOp>(inst)) { supported_block_size = sparse_op.GetFloatBlockSize(); type = mlir::cast<ShapedType>(dyn_cast<ConstOp>(inst).getType()); } else if (isa<QConstOp>(inst)) { supported_block_size = sparse_op.GetQuantizedBlockSize(); type = mlir::cast<ShapedType>(dyn_cast<QConstOp>(inst).getType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
} } const DimensionVector& batch_dimensions() const { return batch_dimensions_; } const DimensionVector& contracting_dimensions() const { return contracting_dimensions_; } // Out dimensions are any dimensions that are neither batch nor contracting // dimensions, hence will be propagated to output shape. const DimensionVector& out_dimensions() const { return out_dimensions_; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc
if (!type) return -1; auto indices_op = indices.getDefiningOp(); if (!indices_op) return -1; auto const_op = llvm::dyn_cast<TF::ConstOp>(indices_op); if (!const_op) return -1; int64_t last_index = -1; int64_t first_index = -1; for (const auto& ind : const_op.getValue().getValues<APInt>()) { if (last_index == -1) { last_index = ind.getSExtValue(); first_index = last_index;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.5K bytes - Viewed (0)