Home
last modified time | relevance | path

Searched defs:node_builder (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.cc103 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in SourceOp() local
111 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in UnaryOp() local
120 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in BinaryOp() local
Dgraph_partition.cc398 NodeBuilder node_builder(node_name, "Enter", g->op_registry()); in AddControlEnter() local
413 NodeBuilder node_builder(node_name, "Merge", g->op_registry()); in AddControlMerge() local
444 NodeBuilder node_builder(options.GetNameForOp("Const"), "Const", in EmptyConst() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Ddecode_csv_vectorizer.cc40 auto node_builder = NodeBuilder(node.type_string(), node.type_string()) in Vectorize() local
Dparse_single_example_vectorizer.cc72 auto node_builder = in Vectorize() local
Dcwise_op_vectorizer.cc120 auto node_builder = NodeBuilder(strings::StrCat("vectorized/", node.name()), in CwiseVectorizeHelper() local
/external/tensorflow/tensorflow/core/common_runtime/
Dparallel_concat_optimizer.cc56 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in Run() local
Daccumulate_n_optimizer.cc79 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in rewriteNode() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc360 NodeBuilder node_builder(opts.GetNameForOp("NoOp"), "NoOp", in Sequencer() local
377 NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const", in KnownShapeBase() local
401 NodeBuilder node_builder(absl::StrCat(call_node, "_key_placeholder"), in KeyPlaceholder() local
418 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaRecvAtHost"), in RecvAtHost() local
437 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaSendFromHost"), in SendFromHost() local
470 NodeBuilder node_builder(opts.GetNameForOp("AddN"), "AddNLikeTest", in AddNLike() local
483 NodeBuilder node_builder(opts.GetNameForOp("Retval"), "_Retval", in RetOp() local
629 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
977 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
1172 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
[all …]
Dencapsulate_subgraphs_pass.cc1844 NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", in AddDummyShapedNode() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc535 NodeDefBuilder node_builder(info.engine_name, "TRTEngineOp"); in CreateTRTNode() local
652 NodeDefBuilder node_builder(StrCat(name, "_Arg"), in RegisterSegmentFunctionToFunctionLibrary() local
679 NodeDefBuilder node_builder(StrCat(name, "_Ret"), in RegisterSegmentFunctionToFunctionLibrary() local
/external/tensorflow/tensorflow/c/
Dc_api_internal.h119 tensorflow::NodeBuilder node_builder; member
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc429 auto node_builder = [](StringPiece op) { in StackTensor() local
Dvectorization_utils_test.cc895 auto node_builder = NodeBuilder("op", op_type); in CwiseTestHelper() local
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc217 NodeBuilder node_builder(opts->GetNameForOp(type_string), type_string, in AddDataset() local