Home
last modified time | relevance | path

Searched defs:buffer_size_node (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dauto_shard.cc265 const string& buffer_size_node, in AddShuffleDataset()
294 const string& buffer_size_node, in AddShuffleDatasetV2()
317 const string& buffer_size_node, in AddShuffleDatasetV3()
367 string* op_name, string* buffer_size_node, in RemoveShuffleDataset()
392 string* op_name, string* buffer_size_node, in RemoveShuffleDatasetV2()
414 string* op_name, string* buffer_size_node, in RemoveShuffleDatasetV3()
443 string buffer_size_node = ""; in ProcessDatasetSourceNode() local
H A Dshuffle_and_repeat_fusion_test.cc55 NodeDef *buffer_size_node = in TEST() local
123 NodeDef *buffer_size_node = in TEST() local
188 NodeDef *buffer_size_node = in TEST() local
H A Dautotune_buffer_sizes.cc71 NodeDef* buffer_size_node = graph.GetNode(node.input(1)); in OptimizeAndCollectStats() local
H A Dslack_test.cc53 NodeDef *buffer_size_node = in SetupGrapplerItem() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dshuffle_dataset_op.cc547 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
607 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
662 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local
911 Node* buffer_size_node = nullptr; in AsGraphDefInternal() local