Searched refs:arg_builder (Results 1 – 4 of 4) sorted by relevance
| /external/rust/crates/clap/src/args/ |
| D | mod.rs | 3 pub use self::arg_builder::{Base, FlagBuilder, OptBuilder, PosBuilder, Switched, Valued};
15 mod arg_builder; module
|
| D | arg.rs | 15 use args::arg_builder::{Base, Switched, Valued};
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 516 NodeDefBuilder arg_builder(absl::StrCat("arg_", arg_idx), "_Arg"); in AddOutsideCompilationInputArgToFunctionBody() local
517 arg_builder.Attr("T", data_type); in AddOutsideCompilationInputArgToFunctionBody()
518 arg_builder.Attr("index", arg_idx); in AddOutsideCompilationInputArgToFunctionBody()
520 TF_RETURN_IF_ERROR(arg_builder.Finalize(&arg_def)); in AddOutsideCompilationInputArgToFunctionBody()
1373 NodeDefBuilder arg_builder("key_arg", FunctionLibraryDefinition::kArgOp); in ReplaceKeyPlaceholderWithArgNode() local
1374 arg_builder.Attr("T", DT_STRING); in ReplaceKeyPlaceholderWithArgNode()
1375 arg_builder.Attr("index", 0); in ReplaceKeyPlaceholderWithArgNode()
1377 TF_RETURN_IF_ERROR(arg_builder.Finalize(&arg_def)); in ReplaceKeyPlaceholderWithArgNode()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 664 NodeDefBuilder arg_builder(absl::StrCat("arg_", i), in MoveHeadOutsideCompilationToHost() local
666 arg_builder.Attr("T", new_arg_types[i - old_num_per_replica_inputs]); in MoveHeadOutsideCompilationToHost()
667 arg_builder.Attr("index", i); in MoveHeadOutsideCompilationToHost()
669 TF_RETURN_IF_ERROR(arg_builder.Finalize(&arg_def)); in MoveHeadOutsideCompilationToHost()
|