Home
last modified time | relevance | path

Searched defs:shuffle_node (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dshuffle_and_repeat_fusion.cc44 Status FuseShuffleV1AndRepeat(const NodeDef& shuffle_node, in FuseShuffleV1AndRepeat()
78 Status FuseShuffleV2AndRepeat(const NodeDef& shuffle_node, in FuseShuffleV2AndRepeat()
118 Status FuseShuffleV3AndRepeat(const NodeDef& shuffle_node, in FuseShuffleV3AndRepeat()
169 const NodeDef& shuffle_node = in OptimizeAndCollectStats() local
H A Dshuffle_and_repeat_fusion_test.cc64 NodeDef *shuffle_node = graph_utils::AddNode( in TEST() local
131 NodeDef *shuffle_node = graph_utils::AddNode( in TEST() local
200 NodeDef *shuffle_node = graph_utils::AddNode( in TEST() local