- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getSizeSplits (0.12 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)