Searched defs:repeat_node (Results 1 – 3 of 3) sorted by relevance
45 const NodeDef& repeat_node, in FuseShuffleV1AndRepeat()79 const NodeDef& repeat_node, in FuseShuffleV2AndRepeat()119 const NodeDef& repeat_node, in FuseShuffleV3AndRepeat()164 for (const NodeDef& repeat_node : item.graph.node()) { in OptimizeAndCollectStats() local
72 NodeDef *repeat_node = graph_utils::AddNode( in TEST() local138 NodeDef *repeat_node = graph_utils::AddNode( in TEST() local208 NodeDef *repeat_node = graph_utils::AddNode( in TEST() local
61 bool IsRepeatOne(const NodeDef& repeat_node, const MutableGraphView& graph) { in IsRepeatOne()