Searched defs:seed_node (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | auto_shard.cc | 266 const string& seed_node, const string& seed2_node, in AddShuffleDataset()
318 const string& seed_node, const string& seed2_node, in AddShuffleDatasetV3()
368 string* seed_node, string* seed2_node, in RemoveShuffleDataset()
415 string* seed_node, string* seed2_node, in RemoveShuffleDatasetV3()
444 string seed_node = ""; in ProcessDatasetSourceNode() local
|
| H A D | shuffle_and_repeat_fusion_test.cc | 57 NodeDef *seed_node = graph_utils::AddScalarConstNode<int64_t>(-1, &graph); in TEST() local
190 NodeDef *seed_node = graph_utils::AddScalarConstNode<int64_t>(-1, &graph); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | shuffle_dataset_op.cc | 548 Node* seed_node = nullptr; in AsGraphDefInternal() local
663 Node* seed_node = nullptr; in AsGraphDefInternal() local
912 Node* seed_node = nullptr; in AsGraphDefInternal() local
|