- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for BuildIntArrayConstOp (0.27 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h
indices_type.getRank() == 1 && updates_type.getRank() == 1 && index_depth == 1 && num_updates == 1) { ImplicitLocOpBuilder builder(loc, rewriter); auto indices_shape = BuildIntArrayConstOp( builder, rewriter, llvm::SmallVector<int64_t>({num_updates, index_depth}), rewriter.getI32Type()); new_indices = rewriter.create<TF::ReshapeOp>( loc,
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/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc
Type type) { Value result_const = builder.create<TF::ConstOp>(rewriter.getIntegerAttr(type, const_value)); return result_const; } Value BuildIntArrayConstOp(ImplicitLocOpBuilder& builder, ConversionPatternRewriter& rewriter, ArrayRef<int64_t> const_value, Type type) { DenseIntElementsAttr const_value_raw;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.1K bytes - Viewed (0)