- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getSizeSplits (0.28 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
LogicalResult UnrollSplitV::matchAndRewrite(TFL::SplitVOp splitv_op, PatternRewriter& rewriter) const { // We need to make sure both splits & split dim are constants. auto splits = splitv_op.getSizeSplits().getDefiningOp(); mlir::DenseIntElementsAttr splits_attr; if (!splits || !matchPattern(splits, m_Constant(&splits_attr))) return failure(); auto split_dim = splitv_op.getSplitDim().getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
auto num_split = rewriter.getI32IntegerAttr(tf_splitv_op.getNumSplit()); rewriter.replaceOpWithNewOp<TFL::SplitVOp>( op, tf_splitv_op.getOutput().getTypes(), tf_splitv_op.getValue(), tf_splitv_op.getSizeSplits(), tf_splitv_op.getSplitDim(), num_split); return success(); } LogicalResult ConvertTFUnpackOp::matchAndRewrite( Operation* op, PatternRewriter& rewriter) const {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// SplitVOp //===----------------------------------------------------------------------===// LogicalResult SplitVOp::verify() { SplitVOp op = *this; auto split_sizes_type = op.getSizeSplits().getType().dyn_cast<RankedTensorType>(); if (!split_sizes_type) return success(); if (split_sizes_type.getRank() != 1 || (!ShapedType::isDynamic(split_sizes_type.getDimSize(0)) &&
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/ir/tfl_ops.cc
// If 'size_splits' is not a constant, there are no other checks. ElementsAttr size_splits_attr; if (!matchPattern(op.getSizeSplits(), m_Constant(&size_splits_attr))) return success(); if (size_splits_attr.getNumElements() != num_splits) { auto size_splits_type = op.getSizeSplits().getType().cast<RankedTensorType>(); RankedTensorType expected_size_splits_type =
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/tf2xla/transforms/legalize_tf.cc
return failure(); // We can only match when the split sizes is a constant int vector. DenseIntElementsAttr split_sizes_attr; if (!matchPattern(op.getSizeSplits(), m_Constant(&split_sizes_attr))) return failure(); // Get each chunck's size along the dimension to split. It may contain // dynamic sizes and we need to update it if so.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)