Searched defs:node_images (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass.cc | 466 const Edge* edge, const std::unordered_map<const Node*, Node*>& node_images, in RecordArg() argument 502 const std::unordered_map<const Node*, Node*>& node_images) { in RecordControlResult() argument 511 const std::unordered_map<const Node*, Node*>& node_images) { in RecordResult() argument 661 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNode() argument 689 std::unordered_map<const Node*, Node*>* node_images) { in CopySubgraphNodes() argument 704 const std::unordered_map<const Node*, Node*>& node_images, in CopySubgraphEdges() argument 778 std::unordered_map<const Node*, Node*> node_images; in SplitIntoSubgraphs() local 820 Graph* graph_out, std::unordered_map<const Node*, Node*>* node_images) { in CopyNodesToOutputGraph() argument 837 const std::unordered_map<const Node*, Node*>& node_images, in AddFunctionCallNodes() argument 848 const std::unordered_map<const Node*, Node*>& node_images, in FindOutputImageOfEdgeSrc() argument [all …]
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 3973 NodeToNodeReplicasMap* node_images, Graph* graph) { in CopyOutsideCompilationNodes() 4018 NodeToNodeReplicasMap* node_images, Graph* graph) { in ReplicateOutsideCompilationNodes() 4066 const NodeToNodeReplicasMap& node_images, in CopyOutsideCompilationEdges() 4194 const NodeToNodeReplicasMap& node_images, in ReplicateOutsideCompilationEdges() 4206 const NodeToNodeReplicasMap& node_images, Graph* graph) { in RemoveOutsideCompilationNodes()
|
| D | encapsulate_tpu_computations_pass.cc | 463 std::map<Node*, std::vector<Node*>> node_images; in MoveHeadOutsideCompilationToHost() local 1114 std::map<Node*, std::vector<Node*>> node_images; in MoveTailOutsideCompilationToHost() local
|