Searched defs:split_op (Results 1 – 8 of 8) sorted by relevance
85 auto split_op = mlir::cast<mlir::TF::SplitOp>(op); in ExpandOp() local102 auto split_op = mlir::cast<mlir::TF::SplitOp>(op); in ComputeLayoutForward() local115 auto split_op = mlir::cast<mlir::TF::SplitOp>(op); in ComputeLayoutBackward() local
50 mlir::TF::SplitOp* split_op) { in CreateSplitOp()171 for (auto split_op : split_ops_for_tiled_input) { in HandleTileShardedInputs() local190 for (auto split_op : split_ops_for_tiled_input) in HandleTileShardedInputs() local
153 SmallVector<Value, 4> SliceOutputs(Operation* split_op, Value input, in SliceOutputs()287 LogicalResult UnrollSplit::matchAndRewrite(TFL::SplitOp split_op, in matchAndRewrite()
129 auto split_op = rewriter.create<TF::SplitOp>( in sliceInput() local
734 Operand& split_op = split_vec->operands[0]; in try_reassign_split_vector() local
99 mlir::OpBuilder* builder, mlir::TF::SplitOp* split_op) { in CreateSplitOp()
1155 tensorflow::NodeDef* split_op = tensorflow_graph->add_node(); in ConvertSplitOperator() local1572 tensorflow::NodeDef* split_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
3373 auto split_op = ops::Split( in TEST_F() local