Home
last modified time | relevance | path

Searched defs:batch_node (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_batch_fusion.cc42 NodeDef MakeMapAndBatchNode(const NodeDef& map_node, const NodeDef& batch_node, in MakeMapAndBatchNode()
123 const NodeDef& batch_node = node; in OptimizeAndCollectStats() local
Dmap_and_batch_fusion_test.cc63 NodeDef *batch_node; in TEST() local
147 NodeDef *batch_node; in TEST() local
231 NodeDef *batch_node; in TEST() local
316 NodeDef *batch_node; in TEST() local
Dbatch_parallelization.cc85 const NodeDef* batch_node = get_batch_node(node); in OptimizeAndCollectStats() local