Searched refs:unformat_replicate (Results 1 – 1 of 1) sorted by relevance
446 auto unformat_replicate = builder.create<tf_device::ReplicateOp>( in HandleReplicateOp() local450 builder.setInsertionPointToEnd(&unformat_replicate.GetBody()); in HandleReplicateOp()454 BlockArgument state = unformat_replicate.GetReplicatedBlockArguments().back(); in HandleReplicateOp()456 unformat_replicate.GetReplicatedBlockArguments().drop_back(1); in HandleReplicateOp()457 auto packed_block_args = unformat_replicate.GetPackedBlockArguments(); in HandleReplicateOp()