Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass.cc516 NodeDefBuilder arg_builder(absl::StrCat("arg_", arg_idx), "_Arg"); in AddOutsideCompilationInputArgToFunctionBody() local
1387 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp); in ReplaceKeyPlaceholderWithArgNode() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc636 NodeDefBuilder arg_builder(absl::StrCat("arg_", i), in MoveHeadOutsideCompilationToHost() local