Searched refs:batch_dataset_v2 (Results 1 – 2 of 2) sorted by relevance
334 Node* batch_dataset_v2; in TEST() local373 .Finalize(&graph, &batch_dataset_v2)); in TEST()376 add_identity_nodes(batch_dataset_v2, graph, identity_nodes); in TEST()
5118 variant_tensor = gen_dataset_ops.batch_dataset_v2(