- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getNumSplit (0.12 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
Operation* op, PatternRewriter& rewriter) const { auto tf_split_op = cast<TF::SplitOp>(op); // Number of splits cannot be negative. auto num_split = rewriter.getI32IntegerAttr(tf_split_op.getNumSplit()); rewriter.replaceOpWithNewOp<TFL::SplitOp>( op, tf_split_op.getOutput().getTypes(), tf_split_op.getSplitDim(), tf_split_op.getValue(), num_split); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0)