Home
last modified time | relevance | path

Searched defs:node_builder (Results 1 – 18 of 18) 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.cc425 NodeBuilder node_builder(node_name, "Enter", g->op_registry()); in AddControlEnter() local
440 NodeBuilder node_builder(node_name, "Merge", g->op_registry()); in AddControlMerge() local
471 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.cc52 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in Run() local
Dlower_if_op.cc139 Status CondBuilder::SetColocationAndFinalize(NodeBuilder node_builder, in SetColocationAndFinalize()
Daccumulate_n_optimizer.cc111 NodeBuilder node_builder(name, op, OpRegistry::Global(), &debug_info); in RewriteIntoTempVariable() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc348 NodeBuilder node_builder(opts.GetNameForOp("NoOp"), "NoOp", in Sequencer() local
365 NodeBuilder node_builder(opts.GetNameForOp("Const"), "Const", in KnownShapeBase() local
389 NodeBuilder node_builder(absl::StrCat(call_node, "_key_placeholder"), in KeyPlaceholder() local
408 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaRecvAtHost"), in RecvAtHost() local
428 NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaSendFromHost"), in SendFromHost() local
461 NodeBuilder node_builder(opts.GetNameForOp("AddN"), "AddNLikeTest", in AddNLike() local
474 NodeBuilder node_builder(opts.GetNameForOp("Retval"), "_Retval", in RetOp() local
622 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
976 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
1183 NodeBuilder node_builder("F1", "F1", lib_def.get()); in TEST() local
[all …]
Dcluster_scoping_pass_test.cc76 NodeBuilder node_builder(name, "Stage", opts.op_registry()); in BuildStageNode() local
Dencapsulate_subgraphs_pass.cc1011 NodeBuilder node_builder(options.GetNameForOp("KnownShape"), "Const", in AddDummyShapedNode() local
Dmark_for_compilation_pass_test.cc1736 NodeBuilder node_builder(name, "Stage", opts.op_registry()); in MakeStageNode() local
/external/tensorflow/tensorflow/c/
Dc_api_internal.h111 tensorflow::NodeBuilder node_builder; member
/external/tensorflow/tensorflow/core/framework/
Ddataset.cc240 NodeBuilder node_builder(opts->GetNameForOp(type_string), type_string, in AddDataset() local
519 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in AddDatasetOrTensorHelper() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc422 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/compiler/tf2tensorrt/convert/
Dconvert_graph.cc467 NodeDefBuilder node_builder(info.engine_name, "TRTEngineOp"); in CreateTRTNode() local
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dserialization.cc930 data::CLNodeBuilder node_builder(*builder); in Encode() local