Searched defs:outside_compilation_nodes (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 503 const std::vector<Node*>& outside_compilation_nodes, Graph* g, Node* n) { in AddEdgesFromOutsideCompilationNodes()
615 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForWhile() local
735 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForIf() local
908 std::vector<Node*> outside_compilation_nodes; in PostprocessLiftedArgsForCall() local
2373 std::vector<Node*> outside_compilation_nodes; in ExtractOutsideCompilationForFunction() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | distributed_tpu_rewrite_pass.cc | 334 std::map<string, DistributedTPURewritePass::OutsideCompilationNodeMap>* in FindTaggedNodes()
3919 int replica_index, const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationNodes()
3964 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationNodes()
4012 const std::vector<Node*>& outside_compilation_nodes, in CopyOutsideCompilationEdges()
4140 const OutsideCompilationNodeMap& outside_compilation_nodes, in ReplicateOutsideCompilationEdges()
4354 const OutsideCompilationNodeMap& outside_compilation_nodes, in ParseHostComputeCores()
4714 const OutsideCompilationNodeMap& outside_compilation_nodes, in RewriteTPUReplicateNode()
4965 std::map<string, OutsideCompilationNodeMap> outside_compilation_nodes; in InternalRun() local
|