Home
last modified time | relevance | path

Searched refs:buffer_size_node (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dauto_shard.cc206 const string& buffer_size_node, in AddShuffleDataset() argument
216 new_node.add_input(buffer_size_node); in AddShuffleDataset()
235 const string& buffer_size_node, in AddShuffleDatasetV2() argument
244 new_node.add_input(buffer_size_node); in AddShuffleDatasetV2()
258 const string& buffer_size_node, in AddShuffleDatasetV3() argument
269 new_node.add_input(buffer_size_node); in AddShuffleDatasetV3()
308 string* op_name, string* buffer_size_node, in RemoveShuffleDataset() argument
313 *buffer_size_node = node.input(1); in RemoveShuffleDataset()
323 graph, *fanin.node, nodes_to_delete, op_name, buffer_size_node, in RemoveShuffleDataset()
333 string* op_name, string* buffer_size_node, in RemoveShuffleDatasetV2() argument
[all …]
Dautotune_buffer_sizes.cc72 NodeDef* buffer_size_node = graph.GetNode(node.input(1)); in OptimizeAndCollectStats() local
74 if (buffer_size_node->op() == "Const") { in OptimizeAndCollectStats()
76 buffer_size_node->attr().at("value").tensor().int64_val(0); in OptimizeAndCollectStats()
79 {buffer_size_node->name(), 0}, in OptimizeAndCollectStats()
Dlatency_all_edges_test.cc49 NodeDef buffer_size_node = NDef("buffer_size_node", "Const", {}, in TEST_P() local
60 buffer_size_node, prefetch_node, model_node}); in TEST_P()
64 buffer_size_node, prefetch_node}); in TEST_P()
Dshuffle_and_repeat_fusion_test.cc55 NodeDef *buffer_size_node = in TEST() local
61 shuffle_inputs[1] = buffer_size_node->name(); in TEST()
123 NodeDef *buffer_size_node = in TEST() local
129 shuffle_inputs[1] = buffer_size_node->name(); in TEST()
188 NodeDef *buffer_size_node = in TEST() local
196 shuffle_inputs[1] = buffer_size_node->name(); in TEST()
Dslack_test.cc53 NodeDef *buffer_size_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in SetupGrapplerItem() local
56 {range_node->name(), buffer_size_node->name()}, common_attrs, &graph); in SetupGrapplerItem()
Dmap_vectorization_test.cc120 auto buffer_size_node = graph_utils::AddScalarConstNode(buffer_size, graph); in AddPrefetchNode() local
123 /*inputs=*/{input_dataset, buffer_size_node->name()}, in AddPrefetchNode()
/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc469 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
474 TF_RETURN_IF_ERROR(b->AddScalar(buffer_size_, &buffer_size_node)); in AsGraphDefInternal()
481 {input_graph_node, buffer_size_node, seed_node, seed2_node}, // Inputs in AsGraphDefInternal()
529 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
530 TF_RETURN_IF_ERROR(b->AddScalar(buffer_size_, &buffer_size_node)); in AsGraphDefInternal()
537 {input_graph_node, buffer_size_node, resource_handle_node}, // Inputs in AsGraphDefInternal()
584 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
587 TF_RETURN_IF_ERROR(b->AddScalar(buffer_size_, &buffer_size_node)); in AsGraphDefInternal()
599 {input_graph_node, buffer_size_node, seed_node, in AsGraphDefInternal()
833 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
[all …]