Searched refs:outside_compilation_node_images (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.h | 556 NodeToNodeReplicasMap* outside_compilation_node_images, Graph* graph,
|
| D | distributed_tpu_rewrite_pass.cc | 4215 NodeToNodeReplicasMap* outside_compilation_node_images, Graph* graph, in RewriteTPUReplicateNode() argument
4363 outside_compilation_node_images, graph)); in RewriteTPUReplicateNode()
4477 NodeToNodeReplicasMap outside_compilation_node_images; in Run() local
4485 &outside_compilation_node_images, graph, shape_info, in Run()
4506 iter.second, outside_compilation_node_images, in Run()
4510 RemoveOutsideCompilationNodes(outside_compilation_node_images, graph)); in Run()
|