- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for slice_v1 (3.16 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
LogicalResult matchAndRewrite( mhlo::SliceOp slice_op, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const final { auto begin = rewriter.create<TF::ConstOp>(slice_op.getLoc(), slice_op.getStartIndices()); auto end = rewriter.create<TF::ConstOp>(slice_op.getLoc(), slice_op.getLimitIndices()); auto strides =
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/tensorflow/ir/tf_ops_n_z.cc
// All masks are `0` except `shrink_axis_mask` which is equal to `1` (slicing // scalar value from input vector). if (slice_op.getBeginMask() != 0 || slice_op.getEllipsisMask() != 0 || slice_op.getEndMask() != 0 || slice_op.getNewAxisMask() != 0 || slice_op.getShrinkAxisMask() != 1) return {}; // Returns a value if the `value` is defined by a ConstOp with a single
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
auto slice_op = llvm::dyn_cast_or_null<TFL::SliceOp>(value.getUses().begin().getUser()); // We only match for the case where value is used by SliceOp. if (!slice_op) return std::nullopt; DenseElementsAttr begin; DenseElementsAttr size; if (!matchPattern(slice_op->getOperand(1), m_Constant(&begin)) || !matchPattern(slice_op->getOperand(2), m_Constant(&size)))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
using OpRewritePattern<TFL::SliceOp>::OpRewritePattern; LogicalResult matchAndRewrite(TFL::SliceOp slice_op, PatternRewriter& rewriter) const override { auto begin = slice_op.getBegin(); auto size = slice_op.getSize(); auto begin_type = begin.getType().dyn_cast_or_null<RankedTensorType>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
tflite::BuiltinOptions2_StablehloConcatenateOptions, concat_option.Union()); } if (auto vhlo_op = llvm::dyn_cast<mlir::vhlo::SliceOpV1>(inst)) { std::string op_name = inst->getName().getStringRef().str(); uint32_t opcode_index = GetOpcodeIndex(op_name, tflite::BuiltinOperator_STABLEHLO_SLICE);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)