- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 37 for ReplaceOpWithNewOp (0.36 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
auto table_id = static_cast<int32_t>(::llvm::hash_value(hashtable_op.getSharedName())); auto key_dtype = hashtable_op.getKeyDtype(); auto value_dtype = hashtable_op.getValueDtype(); rewriter.replaceOpWithNewOp<TFL::HashtableOp>( hashtable_op, output_type, table_id, key_dtype, value_dtype); return success(); } }; class LegalizeHashTableFindOpPattern
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( tfg::ReturnOp ret, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const final { rewriter.replaceOpWithNewOp<tf_executor::FetchOp>(ret.getOperation(), adaptor.getOperands()); return success(); } }; class ConvertControlTriggerOp : public ConversionPattern {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
IntegerAttr axis_i32; if (failed(ConvertToI32Attr(axis_int, &axis_i32))) return failure(); StringAttr fused_activation_function = StringAttr::get(rewriter.getContext(), "NONE"); rewriter.replaceOpWithNewOp<ConcatenationOp>( op, output_type, values, axis_i32, fused_activation_function); return success(); } template <typename BatchMatMulOpType>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
rhs_type.getRank() < 1 || 2 < rhs_type.getRank()) { return rewriter.notifyMatchFailure( op, "unsupported dot operation type; operands must be vectors or " "matrices"); } rewriter.replaceOpWithNewOp<mhlo::DotGeneralOp>( op, op.getType(), op.getLhs(), op.getRhs(), mhlo::DotDimensionNumbersAttr::get( op.getContext(), /*lhsBatchingDimensions=*/{}, /*rhsBatchingDimensions=*/{},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 26.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc
PatternRewriter& rewriter) const override { Value value = nullptr; for (auto v : op.getComponents()) { value = v; } if (!value) return failure(); rewriter.replaceOpWithNewOp<TF::IdentityOp>(op, value.getType(), value); return success(); } }; class HandleOptionalGet : public OpRewritePattern<TF::OptionalGetValueOp> { public: using OpRewritePattern::OpRewritePattern;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
return success(); } void rewrite(stablehlo::UniformQuantizeOp op, PatternRewriter& rewriter) const override { Type output_type = *op->getResultTypes().begin(); rewriter.replaceOpWithNewOp<TFL::QuantizeOp>( op, output_type, /*input=*/op.getOperand(), /*qtype=*/TypeAttr::get(output_type)); } }; // stablehlo.uniform_dequantize -> tfl.dequantize
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/post_quantize.cc
if (auto qtype = quant::QuantizedType::getQuantizedElementType( q.getInput().getType())) { rewriter.setInsertionPoint(op); rewriter.replaceOpWithNewOp<DequantizeOp>(op, op.getOutput().getType(), q.getInput()); return success(); } op.replaceAllUsesWith(q.getInput());
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/lite/transforms/lower_static_tensor_list.cc
tensorflow::TensorShape(result_shape)); auto attr_or = tensorflow::ConvertTensor(tensor, &rewriter); if (!attr_or.ok()) return failure(); rewriter.replaceOpWithNewOp<TF::ConstOp>(op, attr_or.value()); return success(); } // Extract individual tensor list element and combine them using the tf.Pack // op. Location loc = op.getLoc();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0)