Home
last modified time | relevance | path

Searched defs:pivot_node (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass.cc1129 Node* pivot_node) { in ExpandHostGraphIntoMainGraph()
1209 Graph* host_graph, Node* pivot_node, in RewriteShapeInferenceGraph()
2531 Node* pivot_node = node_name_index[pivot_name]; in ExtractOutsideCompilation() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc334 Node* pivot_node) { in MoveHeadOutsideCompilationToHost()
970 Node* pivot_node) { in MoveTailOutsideCompilationToHost()
2844 Node* pivot_node = pivots[iter.first]; in ProcessHeadTailOutsideCompilation() local