Home
last modified time | relevance | path

Searched defs:batch_node (Results 1 – 4 of 4) 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()
121 const NodeDef& batch_node = node; in OptimizeAndCollectStats() local
Dmap_and_batch_fusion_test.cc64 NodeDef *batch_node; in TEST() local
149 NodeDef *batch_node; in TEST() local
234 NodeDef *batch_node; in TEST() local
320 NodeDef *batch_node; in TEST() local
Dbatch_parallelization.cc84 const NodeDef* batch_node = get_batch_node(node); in OptimizeAndCollectStats() local
Dauto_shard.cc511 const NodeDef& batch_node) { in GetDropRemainder()