- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for ConversionPatternRewriter (0.29 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/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.cc
mhlo::ReduceOp reduce_op, Value input, TF::ConstOp reduction_indices, ConversionPatternRewriter& rewriter) { return failure(); } template <> LogicalResult rewriteNonMatchInitValue<TF::MinOp, void>( mhlo::ReduceOp reduce_op, Value input, TF::ConstOp reduction_indices, ConversionPatternRewriter& rewriter) { return failure(); }
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/transforms/lower_static_tensor_list.cc
struct ConvertConst : public OpConversionPattern<TF::ConstOp> { using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( TF::ConstOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { // Verify that the tensor proto contains tensor of type variant and scalar // shape. The variant type should hold a TensorList.
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/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/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.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)