Searched defs:outside_compilation_nodes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 503 const std::vector<Node*>& outside_compilation_nodes, Graph* g, Node* n) { in AddEdgesFromOutsideCompilationNodes()
619 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForWhile() local
735 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForIf() local
901 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForCall() local
2371 std::vector<Node*> outside_compilation_nodes; in ExtractOutsideCompilationForFunction() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 331 std::map<string, DistributedTPURewritePass::OutsideCompilationNodeMap>* in FindTaggedNodes()
3970 int replica_index, const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationNodes()
4017 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationNodes()
4065 const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationEdges()
4193 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationEdges()
4407 const OutsideCompilationNodeMap& outside_compilation_nodes, in ParseHostComputeCores()
4771 const OutsideCompilationNodeMap& outside_compilation_nodes, in RewriteTPUReplicateNode()
5014 std::map<string, OutsideCompilationNodeMap> outside_compilation_nodes; in Run() local
|