Searched defs:buffer_size_node (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | auto_shard.cc | 218 const string& buffer_size_node, in AddShuffleDataset()
247 const string& buffer_size_node, in AddShuffleDatasetV2()
270 const string& buffer_size_node, in AddShuffleDatasetV3()
320 string* op_name, string* buffer_size_node, in RemoveShuffleDataset()
345 string* op_name, string* buffer_size_node, in RemoveShuffleDatasetV2()
367 string* op_name, string* buffer_size_node, in RemoveShuffleDatasetV3()
396 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 | autotune_buffer_sizes.cc | 71 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
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 493 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
553 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
608 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
857 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
|