- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for ConversionPatternRewriter (0.2 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.h
PermutationAndShape GetPermutationAndTransposedShape( llvm::ArrayRef<int64_t> permutation_array, ShapedType input_type, ConversionPatternRewriter& rewriter); // Create a single const integer. Value BuildIntConstOp(ImplicitLocOpBuilder& builder, ConversionPatternRewriter& rewriter, int64_t const_value, Type type); // Create a const integer vector tensor (1-dim).
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 08 11:35:25 UTC 2023 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/custom_call.h
public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( mhlo::CustomCallOp mhlo_custom_call, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const final; }; // Ops that have a call_target_name starting with the prefix "custom_call." and // backend_config of type StringAttr (if specified) should be legalized (i.e.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 02:41:24 UTC 2023 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/custom_call.cc
#include "xla/mlir_hlo/mhlo/IR/hlo_ops.h" namespace mlir { namespace odml { LogicalResult ConvertCustomCallOp::matchAndRewrite( mhlo::CustomCallOp mhlo_custom_call, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const { auto tfl_custom = rewriter.create<TFL::CustomOp>( mhlo_custom_call.getLoc(), mhlo_custom_call.getResultTypes(), mhlo_custom_call.getInputs()); tfl_custom.setCustomCodeAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.cc
LogicalResult CanonicalizeScatterUpdates( Operation* scatter_op, llvm::ArrayRef<int64_t> update_window_dims, const Value& indices, const ShapedType& indices_type, Value& updates, ShapedType& updates_type, ConversionPatternRewriter& rewriter) { auto canonical_update_window_dims = llvm::to_vector( llvm::seq<int64_t>(indices_type.getRank() - 1, updates_type.getRank()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Sep 12 02:29:42 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
: ConversionPattern(converter, MatchAnyOpTypeTag(), /*benefit=*/1, ctx) {} LogicalResult matchAndRewrite( Operation* op, const ArrayRef<Value> operands, ConversionPatternRewriter& rewriter) const override { if (getTypeConverter()->isLegal(op)) { return failure(); } if (isa<mlir::stablehlo::BitcastConvertOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_with_tf2xla.cc
device_type_(device_type), prefer_tf2xla_(prefer_tf2xla) {} LogicalResult matchAndRewrite( Operation* op, ArrayRef<Value> operands, ConversionPatternRewriter& rewriter) const override { // This pattern is a conversion pattern because we want to specify a type // converter. However, this pattern still uses the original op's operands
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/reduce.h
public: using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( mhlo::ReduceOp reduce_op, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const final { if (reduce_op.getInputs().size() != 2) return failure(); if (reduce_op.getDimensions().getNumElements() != 1) return failure(); // Check that the operand init is the expected value.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.7K bytes - Viewed (0)