Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass.cc531 NodeDefBuilder ret_builder(absl::StrCat("ret_", arg_idx), "_Retval"); in AddMatchingRetvalNode() local
1610 NodeDefBuilder ret_builder( in RewriteHostWhileLoopCond() local
1668 NodeDefBuilder ret_builder( in RewriteHostWhileLoopBody() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc1152 NodeDefBuilder ret_builder(absl::StrCat("ret_", i), in MoveTailOutsideCompilationToHost() local