Searched defs:seed2_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()
445 string seed2_node = ""; in ProcessDatasetSourceNode() local
|
| H A D | shuffle_and_repeat_fusion_test.cc | 58 NodeDef *seed2_node = graph_utils::AddScalarConstNode<int64_t>(-1, &graph); in TEST() local
191 NodeDef *seed2_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 | 549 Node* seed2_node = nullptr; in AsGraphDefInternal() local
664 Node* seed2_node = nullptr; in AsGraphDefInternal() local
913 Node* seed2_node = nullptr; in AsGraphDefInternal() local
|