- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for constant_ops (0.14 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
class RewriteQuantizedConstantOp : public OpRewritePattern<stablehlo::ConstantOp> { public: using OpRewritePattern<stablehlo::ConstantOp>::OpRewritePattern; LogicalResult match(stablehlo::ConstantOp op) const override { return success(IsQuantizedTensorType(op.getOutput().getType())); } void rewrite(stablehlo::ConstantOp op, PatternRewriter& rewriter) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
added_value.reshape(RankedTensorType::get( {num_dims}, mlir::cast<ShapedType>(added_value.getType()).getElementType())); ::mlir::arith::ConstantOp new_end = rewriter.create<arith::ConstantOp>( strided_slice_op.getEnd().getLoc(), new_added_value); if (strided_slice_op.getBeginMask() != 0) return failure(); if (strided_slice_op.getEndMask() != 0) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
{pad_op.getEdgePaddingLow().size()}, rewriter.getI64Type()); auto slice_begins_const_op = rewriter.create<arith::ConstantOp>( loc, slice_attr_type, DenseIntElementsAttr::get(slice_attr_type, slice_begins)); auto slice_sizes_const_op = rewriter.create<arith::ConstantOp>( loc, slice_attr_type, DenseIntElementsAttr::get(slice_attr_type, slice_sizes));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
} static bool IsConst(Operation* op) { return isa<mlir::func::ConstantOp, mlir::arith::ConstantOp, mlir::TF::ConstOp, tfl::ConstOp, tfl::QConstOp, tfl::SparseConstOp, tfl::SparseQConstOp, mlir::TFL::NoValueOp, mlir::stablehlo::ConstantOp, mlir::vhlo::ConstantOpV1>(op); } static bool IsTFResourceOp(Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
LogicalResult matchAndRewrite(ConstOp const_op, PatternRewriter& rewriter) const override { if (arith::ConstantOp::isBuildableWith(const_op.getValue(), const_op.getType())) { rewriter.replaceOpWithNewOp<arith::ConstantOp>(const_op, const_op.getValue()); return success();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0)