- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for ConversionPatternRewriter (1.38 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
ConversionPatternRewriter& rewriter, int64_t const_value, Type type) { Value result_const = builder.create<TF::ConstOp>(rewriter.getIntegerAttr(type, const_value)); return result_const; } Value BuildIntArrayConstOp(ImplicitLocOpBuilder& builder, ConversionPatternRewriter& rewriter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
class ConvertGraphOp : public OpConversionPattern<tfg::GraphOp> { public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( tfg::GraphOp graph, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const final { Location loc = graph.getLoc(); // To keep the import-as-graph logic taken by TFG, we create `void func()` // to contain the ops in the tfg::GraphOp. That means the arguments/results
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/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( TF::UniformQuantizedDotHybridOp op, TF::UniformQuantizedDotHybridOpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { // Uniform Quantized type for the rhs. int64_t rhs_quantized_dimension = op.getRhsQuantizationAxis(); // Currently for dot, PTQ supports per-tensor quantization.
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/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h
ShapedType& updates_type, ConversionPatternRewriter& rewriter); template <typename BinaryOp, typename TfOp> class ConvertScatterOp : public OpConversionPattern<mhlo::ScatterOp> { public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( mhlo::ScatterOp scatter_op, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const final { {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
: ConversionPattern(converter, MatchAnyOpTypeTag(), 1, ctx) {} LogicalResult matchAndRewrite( Operation *op, ArrayRef<Value> operands, ConversionPatternRewriter &rewriter) const override { // This pattern only handle non-UQ, non-const ops. if (IsTFUniformQuantizedOp(op) || llvm::isa<TF::ConstOp>(op)) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0)