Searched defs:batch_node (Results 1 – 5 of 5) sorted by relevance
42 NodeDef MakeMapAndBatchNode(const NodeDef& map_node, const NodeDef& batch_node, in MakeMapAndBatchNode()123 const NodeDef& batch_node = node; in OptimizeAndCollectStats() local
63 NodeDef *batch_node; in TEST() local147 NodeDef *batch_node; in TEST() local231 NodeDef *batch_node; in TEST() local316 NodeDef *batch_node; in TEST() local
182 const NodeDef& batch_node = in CheckNotVectorized() local413 auto batch_node = AddBatchNode(&graph, map_node->name(), true); in TEST_P() local524 auto batch_node = AddBatchNode(&graph, map_node->name()); in TEST() local543 auto batch_node = AddBatchNode(&graph, map_node->name()); in TEST() local562 auto batch_node = AddBatchNode(&graph, map_node->name()); in TEST() local580 auto batch_node = AddBatchNode(&graph, map_node->name()); in TEST() local
87 const NodeDef* batch_node = get_batch_node(node); in OptimizeAndCollectStats() local
198 NodeDef batch_node; in AddNewBatchNode() local475 const NodeDef*& batch_node = *batch_node_output; in FindMapAndBatchPattern() local546 const NodeDef* batch_node; in OptimizeAndCollectStats() local