Searched defs:buffer_size_node (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 206 const string& buffer_size_node, in AddShuffleDataset()
235 const string& buffer_size_node, in AddShuffleDatasetV2()
258 const string& buffer_size_node, in AddShuffleDatasetV3()
308 string* op_name, string* buffer_size_node, in RemoveShuffleDataset()
333 string* op_name, string* buffer_size_node, in RemoveShuffleDatasetV2()
355 string* op_name, string* buffer_size_node, in RemoveShuffleDatasetV3()
384 string buffer_size_node = ""; in ProcessDatasetSourceNode() local
|
| D | shuffle_and_repeat_fusion_test.cc | 55 NodeDef *buffer_size_node = in TEST() local
123 NodeDef *buffer_size_node = in TEST() local
188 NodeDef *buffer_size_node = in TEST() local
|
| D | latency_all_edges_test.cc | 49 NodeDef buffer_size_node = NDef("buffer_size_node", "Const", {}, in TEST_P() local
|
| D | autotune_buffer_sizes.cc | 72 NodeDef* buffer_size_node = graph.GetNode(node.input(1)); in OptimizeAndCollectStats() local
|
| D | slack_test.cc | 53 NodeDef *buffer_size_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in SetupGrapplerItem() local
|
| D | map_vectorization_test.cc | 120 auto buffer_size_node = graph_utils::AddScalarConstNode(buffer_size, graph); in AddPrefetchNode() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 469 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
529 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
584 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
833 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
|