Searched refs:rebatch_dataset_v2 (Results 1 – 2 of 2) sorted by relevance
340 Node* rebatch_dataset_v2; in TEST_F() local381 .Finalize(&graph, &rebatch_dataset_v2)); in TEST_F()383 add_identity_nodes(rebatch_dataset_v2, graph, identity_nodes); in TEST_F()
151 variant_tensor = ged_ops.rebatch_dataset_v2(