Searched refs:tf_splitv_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_tf.cc | 317 auto tf_splitv_op = cast<TF::SplitVOp>(op); in matchAndRewrite() local
320 auto num_split = rewriter.getI32IntegerAttr(tf_splitv_op.num_split()); in matchAndRewrite()
323 op, tf_splitv_op.output().getTypes(), tf_splitv_op.value(), in matchAndRewrite()
324 tf_splitv_op.size_splits(), tf_splitv_op.split_dim(), num_split); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1570 auto tf_splitv_op = cast<TF::SplitVOp>(op); in matchAndRewrite() local
1575 if (!matchPattern(tf_splitv_op.size_splits(), in matchAndRewrite()
1586 if (!matchPattern(tf_splitv_op.split_dim(), m_Constant(&axisAttrElems))) { in matchAndRewrite()
1593 convertSplitVOp(rewriter, op, tf_splitv_op.getResult(0), in matchAndRewrite()
1594 tf_splitv_op.value(), size_split, axis); in matchAndRewrite()
|