Searched refs:rebatch_dataset (Results 1 – 2 of 2) sorted by relevance
289 Node* rebatch_dataset; in TEST_F() local323 .Finalize(&graph, &rebatch_dataset)); in TEST_F()325 add_identity_nodes(rebatch_dataset, graph, identity_nodes); in TEST_F()
305 variant_tensor = ged_ops.rebatch_dataset(