Searched refs:seed_node (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 207 const string& seed_node, const string& seed2_node, in AddShuffleDataset() argument
217 new_node.add_input(seed_node); in AddShuffleDataset()
259 const string& seed_node, const string& seed2_node, in AddShuffleDatasetV3() argument
270 new_node.add_input(seed_node); in AddShuffleDatasetV3()
309 string* seed_node, string* seed2_node, in RemoveShuffleDataset() argument
314 *seed_node = node.input(2); in RemoveShuffleDataset()
324 seed_node, seed2_node, reshuffle_each_iteration)); in RemoveShuffleDataset()
356 string* seed_node, string* seed2_node, in RemoveShuffleDatasetV3() argument
362 *seed_node = node.input(2); in RemoveShuffleDatasetV3()
373 seed_node, seed2_node, seed_generator_node, reshuffle_each_iteration)); in RemoveShuffleDatasetV3()
[all …]
|
| D | shuffle_and_repeat_fusion_test.cc | 57 NodeDef *seed_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST() local
62 shuffle_inputs[2] = seed_node->name(); in TEST()
190 NodeDef *seed_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST() local
197 shuffle_inputs[2] = seed_node->name(); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 470 Node* seed_node = nullptr; in AsGraphDefInternal() local
475 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed_node)); in AsGraphDefInternal()
481 {input_graph_node, buffer_size_node, seed_node, seed2_node}, // Inputs in AsGraphDefInternal()
585 Node* seed_node = nullptr; in AsGraphDefInternal() local
588 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed_node)); in AsGraphDefInternal()
599 {input_graph_node, buffer_size_node, seed_node, in AsGraphDefInternal()
834 Node* seed_node = nullptr; in AsGraphDefInternal() local
838 TF_RETURN_IF_ERROR(b->AddScalar(seeds_.input_seed(), &seed_node)); in AsGraphDefInternal()
850 {input_graph_node, buffer_size_node, seed_node, in AsGraphDefInternal()
|