Home
last modified time | relevance | path

Searched refs:NodeBuilder (Results 1 – 25 of 151) sorted by relevance

1234567

/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc25 NodeBuilder::NodeOut::NodeOut(Node* n, int32 i) // NOLINT(runtime/explicit) in NodeOut()
32 NodeBuilder::NodeOut::NodeOut(OutputTensor t) : NodeOut(t.node, t.index) {} in NodeOut()
34 NodeBuilder::NodeOut::NodeOut(StringPiece n, int32 i, DataType t) in NodeOut()
37 NodeBuilder::NodeOut::NodeOut() in NodeOut()
40 NodeBuilder::NodeBuilder(StringPiece name, StringPiece op_name, in NodeBuilder() function in tensorflow::NodeBuilder
45 NodeBuilder::NodeBuilder(StringPiece name, const OpDef* op_def) in NodeBuilder() function in tensorflow::NodeBuilder
48 NodeBuilder::NodeBuilder(const NodeDefBuilder& def_builder) in NodeBuilder() function in tensorflow::NodeBuilder
51 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) { in Input()
60 NodeBuilder& NodeBuilder::Input(NodeOut src) { in Input()
70 NodeBuilder& NodeBuilder::Input(gtl::ArraySlice<NodeOut> src_list) { in Input()
[all …]
Dnode_builder.h42 class NodeBuilder {
79 NodeBuilder(StringPiece name, StringPiece op_name,
82 NodeBuilder(StringPiece name, const OpDef* op_def);
85 NodeBuilder(const NodeDefBuilder& def_builder);
91 NodeBuilder& Input(Node* src_node, int src_index = 0);
92 NodeBuilder& Input(NodeOut src);
95 NodeBuilder& Input(gtl::ArraySlice<NodeOut> src_list);
98 NodeBuilder& ControlInput(Node* src_node);
99 NodeBuilder& ControlInputs(gtl::ArraySlice<Node*> src_nodes);
103 NodeBuilder& Device(StringPiece device_spec);
[all …]
Dtestlib.cc38 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "_Send") in Send()
55 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "_Recv") in Recv()
68 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Const") in Constant()
77 TF_CHECK_OK(NodeBuilder(name, "Const") in Constant()
90 TF_CHECK_OK(NodeBuilder(name, "HostConst") in HostConstant()
99 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Variable") in Var()
109 TF_CHECK_OK(NodeBuilder(name, "Variable") in Var()
118 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Assign") in Assign()
128 TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Cumsum") in Cumsum()
140 TF_CHECK_OK(NodeBuilder(g->NewName("n"), reduce, g->op_registry()) in Reduce()
[all …]
Dnode_builder_test.cc33 TF_EXPECT_OK(NodeBuilder("source_op", "Source") in TEST()
39 TF_EXPECT_OK(NodeBuilder("sink1", "Sink") in TEST()
42 TF_EXPECT_OK(NodeBuilder("sink2", "Sink") in TEST()
47 EXPECT_FALSE(NodeBuilder("sink3", "Sink") in TEST()
51 EXPECT_FALSE(NodeBuilder("sink4", "Sink") in TEST()
55 EXPECT_FALSE(NodeBuilder("sink5", "Sink") in TEST()
62 EXPECT_FALSE(NodeBuilder("sink6", "Sink") in TEST()
66 EXPECT_FALSE(NodeBuilder("sink7", "Sink") in TEST()
67 .Input(NodeBuilder::NodeOut(nullptr, 0)) in TEST()
Dsubgraph.h59 virtual Status AddNode(Graph* g, NodeBuilder::NodeOut tensor,
125 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor,
136 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor,
146 Status AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor,
158 Status AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor,
/external/tensorflow/tensorflow/compiler/tf2xla/
Dresource_util_test.cc95 NodeBuilder stack_size_placeholder_builder("stack_size", "Placeholder", in TEST()
101 NodeBuilder stack_op_builder("stack_op", "StackV2", op_reg); in TEST()
105 NodeBuilder stack_close_builder("stack_close", "StackCloseV2", op_reg); in TEST()
130 NodeBuilder stack_size_placeholder_builder("stack_size", "Placeholder", in TEST()
136 NodeBuilder stack_op_builder("stack_op", "StackV2", op_reg); in TEST()
140 NodeBuilder resource_identity_builder("resource_identity", "Identity", in TEST()
145 NodeBuilder stack_close_builder("stack_close", "StackCloseV2", op_reg); in TEST()
174 NodeBuilder stack_size_placeholder_builder("stack_size", "Placeholder", in TEST()
180 NodeBuilder stack_op_builder("stack_op", "StackV2", op_reg); in TEST()
184 NodeBuilder stack_close0_builder("stack_close0", "StackCloseV2", op_reg); in TEST()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dquantize_training.cc201 NodeBuilder save_op_builder = in ConnectVariablesToSaveOp()
202 NodeBuilder(save_op->name(), save_op->type_string()); in ConnectVariablesToSaveOp()
207 std::vector<NodeBuilder::NodeOut> var_nodeouts; in ConnectVariablesToSaveOp()
277 TF_RETURN_IF_ERROR(NodeBuilder(tensor_names_op_name, "Const") in AddRestoreVariableSubgraphs()
286 TF_RETURN_IF_ERROR(NodeBuilder(shape_and_slices_op_name, "Const") in AddRestoreVariableSubgraphs()
293 TF_RETURN_IF_ERROR(NodeBuilder(new_restore_op_name, "RestoreV2") in AddRestoreVariableSubgraphs()
302 TF_RETURN_IF_ERROR(NodeBuilder(new_assign_op_name, "Assign") in AddRestoreVariableSubgraphs()
338 NodeBuilder(strings::StrCat(name_prefix, "/RangeStart"), "Const") in MakeReductionAxes()
346 NodeBuilder(strings::StrCat(name_prefix, "/RangeDelta"), "Const") in MakeReductionAxes()
352 NodeBuilder(strings::StrCat(name_prefix, "/InputRank"), "Rank") in MakeReductionAxes()
[all …]
Dlower_if_op.cc26 using NodeOut = NodeBuilder::NodeOut;
70 Status SetColocationAndFinalize(NodeBuilder node_builder, Graph* graph,
109 NodeBuilder then_call_builder_;
110 NodeBuilder else_call_builder_;
139 Status CondBuilder::SetColocationAndFinalize(NodeBuilder node_builder, in SetColocationAndFinalize()
153 SetColocationAndFinalize(NodeBuilder(NewName("switch_pred"), "Switch", in CreatePivotNodes()
161 SetColocationAndFinalize(NodeBuilder(NewName("pivot_f"), "Identity", in CreatePivotNodes()
167 SetColocationAndFinalize(NodeBuilder(NewName("pivot_t"), "Identity", in CreatePivotNodes()
195 NodeBuilder(NewName(src->name()), "Switch", graph_->op_registry(), in AddInput()
240 NodeBuilder(NewName("output"), "Merge", graph_->op_registry(), in AddOutputs()
[all …]
Dlower_while_op.cc28 using NodeOut = NodeBuilder::NodeOut;
155 NodeBuilder cond_call_builder_;
156 NodeBuilder body_call_builder_;
241 NodeBuilder builder = in CreateEnterNodes()
242 NodeBuilder(NewName("enter"), "Enter", graph_->op_registry(), in CreateEnterNodes()
265 TF_RETURN_IF_ERROR(NodeBuilder(NewName("LoopControlInputs"), "NoOp", in CreateEnterNodes()
284 NodeBuilder(NewName("merge"), "Merge", graph_->op_registry(), in CreateMergeNodes()
310 TF_RETURN_IF_ERROR(NodeBuilder(NewName("LoopCond"), "LoopCond", in CreateCondFuncCallNode()
335 TF_RETURN_IF_ERROR(NodeBuilder(NewName(op_name), op_type, in CreateSwitchNodes()
370 TF_RETURN_IF_ERROR(NodeBuilder(NewName("loop_body_control"), op_type, in CreateBodyFuncCallNode()
[all …]
Dlower_case_op.cc28 using NodeOut = NodeBuilder::NodeOut;
95 std::vector<NodeBuilder> branch_call_builders_;
122 TF_RETURN_IF_ERROR(NodeBuilder(NewName("branch_index"), "_SwitchN", in CreatePivotNodes()
132 TF_RETURN_IF_ERROR(NodeBuilder(NewName(strings::StrCat("pivot_", b)), in CreatePivotNodes()
158 TF_RETURN_IF_ERROR(NodeBuilder(NewName(src->name()), "_SwitchN", in AddInput()
209 TF_RETURN_IF_ERROR(NodeBuilder(NewName("merge"), "Merge", in AddOutputs()
231 TF_RETURN_IF_ERROR(NodeBuilder(NewName("branch_executed"), "Merge", in AddOutputs()
260 NodeBuilder builder = keep_node_fetchable_ && !outputs_.empty() in BuildLoweredCaseOutput()
261 ? NodeBuilder(name_, "IdentityN").Input(outputs_) in BuildLoweredCaseOutput()
262 : NodeBuilder(name_, "NoOp"); in BuildLoweredCaseOutput()
Dlower_while_op_test.cc69 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
75 NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
185 TF_CHECK_OK(NodeBuilder("placed_node", "Placeholder") in TEST()
191 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(placeholder)}); in TEST()
197 NodeBuilder("while", "While", &graph->flib_def()) in TEST()
284 TF_CHECK_OK(NodeBuilder("placed_node", "Placeholder") in TEST()
291 TF_CHECK_OK(NodeBuilder("control_in", "Placeholder") in TEST()
296 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(gpu_0_ph)}); in TEST()
302 NodeBuilder("while", "While", &graph->flib_def()) in TEST()
320 TF_ASSERT_OK(NodeBuilder("control_out", "Const") in TEST()
[all …]
Dshape_refiner_test.cc80 TF_ASSERT_OK(NodeBuilder("test", "TensorAsShapeInt32") in TestStridedSlice()
212 TF_ASSERT_OK(NodeBuilder("b", "TestOpWithNoShapeFn") in TEST_F()
286 TF_ASSERT_OK(NodeBuilder("MI", "MultiIdentity") in TEST_F()
287 .Input(std::vector<NodeBuilder::NodeOut>{small.node(), in TEST_F()
293 TF_ASSERT_OK(NodeBuilder("Test", "ShapeVectorForAllElements") in TEST_F()
299 TF_ASSERT_OK(NodeBuilder("Test", "ShapeVectorForAllElements") in TEST_F()
353 TF_ASSERT_OK(NodeBuilder("Test", "TestOp") in TEST_F()
468 TF_ASSERT_OK(NodeBuilder("Test", "ShapeData") in TEST_F()
502 TF_ASSERT_OK(NodeBuilder("Test", "ShapeDataInt64") in TEST_F()
535 TF_ASSERT_OK(NodeBuilder("Test", "ShapeData") in TEST_F()
[all …]
Dlower_function_call_op_test.cc94 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
95 TF_ASSERT_OK(NodeBuilder("F", "PartitionedCall", &root.graph()->flib_def()) in TEST()
157 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
160 TF_ASSERT_OK(NodeBuilder("B", "PartitionedCall", &root.graph()->flib_def()) in TEST()
169 TF_ASSERT_OK(NodeBuilder("C", "PartitionedCall", &root.graph()->flib_def()) in TEST()
Dlower_functional_ops_test.cc154 std::vector<NodeBuilder::NodeOut> inputs( in TEST()
155 {NodeBuilder::NodeOut(counter.node()), NodeBuilder::NodeOut(pred.node()), in TEST()
156 NodeBuilder::NodeOut(a.node())}); in TEST()
162 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
261 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
267 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
Dlower_if_op_test.cc78 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
80 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
197 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(var.node())}); in TEST()
199 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
281 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
287 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
Dmkl_layout_pass.cc832 std::function<void(const Node*, NodeBuilder*, bool)> copy_attrs;
872 std::function<void(const Node*, NodeBuilder* nb, bool)>)>
874 std::function<void(const Node*, NodeBuilder* nb, bool)> copy_attrs;
1375 std::function<void(const Node*, NodeBuilder* nb, bool)> copy_attrs,
1892 std::vector<NodeBuilder::NodeOut>* output_nodes);
1913 std::vector<NodeBuilder::NodeOut>* output_nodes);
1948 NodeBuilder* nb, const Node* old_node,
1949 std::vector<NodeBuilder::NodeOut>* workspace_tensors,
1962 NodeBuilder* nb, const Node* orig_node);
1973 NodeBuilder* nb);
[all …]
/external/tensorflow/tensorflow/cc/ops/
Dconst_op.cc30 auto builder = NodeBuilder(unique_name, "Const") in ConstHelper()
56 NodeBuilder::NodeOut AsNodeOut(const Scope& scope, const Input& inp) { in AsNodeOut()
59 return NodeBuilder::NodeOut(inp.node(), inp.index()); in AsNodeOut()
62 return NodeBuilder::NodeOut(inp.node(), inp.index()); in AsNodeOut()
65 return NodeBuilder::NodeOut(inp.node_name(), inp.index(), inp.data_type()); in AsNodeOut()
69 return NodeBuilder::NodeOut{transformed.node(), transformed.index()}; in AsNodeOut()
72 std::vector<NodeBuilder::NodeOut> AsNodeOutList(const Scope& scope, in AsNodeOutList()
74 std::vector<NodeBuilder::NodeOut> out; in AsNodeOutList()
/external/clang/include/clang/StaticAnalyzer/Core/PathSensitive/
DCoreEngine.h32 class NodeBuilder; variable
45 friend class NodeBuilder; variable
210 class NodeBuilder {
247 NodeBuilder(ExplodedNode *SrcNode, ExplodedNodeSet &DstSet,
253 NodeBuilder(const ExplodedNodeSet &SrcSet, ExplodedNodeSet &DstSet,
260 virtual ~NodeBuilder() {} in ~NodeBuilder()
312 class NodeBuilderWithSinks: public NodeBuilder {
321 : NodeBuilder(Pred, DstSet, Ctx), Location(L) {} in NodeBuilderWithSinks()
327 return NodeBuilder::generateNode(LocalLoc, State, Pred);
333 ExplodedNode *N = NodeBuilder::generateSink(LocalLoc, State, Pred);
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dsdca_ops_test.cc160 std::vector<NodeBuilder::NodeOut> sparse_indices; in GetGraphs()
161 std::vector<NodeBuilder::NodeOut> sparse_weights; in GetGraphs()
164 NodeBuilder::NodeOut(SparseIndices(g, sparse_features_per_group))); in GetGraphs()
165 sparse_weights.push_back(NodeBuilder::NodeOut(n)); in GetGraphs()
167 std::vector<NodeBuilder::NodeOut> dense_weights; in GetGraphs()
170 dense_weights.push_back(NodeBuilder::NodeOut(n)); in GetGraphs()
173 std::vector<NodeBuilder::NodeOut> sparse_example_indices; in GetGraphs()
174 std::vector<NodeBuilder::NodeOut> sparse_feature_indices; in GetGraphs()
175 std::vector<NodeBuilder::NodeOut> sparse_values; in GetGraphs()
178 sparse_example_indices.push_back(NodeBuilder::NodeOut( in GetGraphs()
[all …]
/external/llvm-project/clang/include/clang/StaticAnalyzer/Core/PathSensitive/
DCoreEngine.h60 friend class NodeBuilder; variable
236 class NodeBuilder {
273 NodeBuilder(ExplodedNode *SrcNode, ExplodedNodeSet &DstSet,
279 NodeBuilder(const ExplodedNodeSet &SrcSet, ExplodedNodeSet &DstSet,
286 virtual ~NodeBuilder() = default;
341 class NodeBuilderWithSinks: public NodeBuilder {
351 : NodeBuilder(Pred, DstSet, Ctx), Location(L) {} in NodeBuilderWithSinks()
357 return NodeBuilder::generateNode(LocalLoc, State, Pred);
363 ExplodedNode *N = NodeBuilder::generateSink(LocalLoc, State, Pred);
378 class StmtNodeBuilder: public NodeBuilder {
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dparse_single_example_vectorizer.cc32 NodeBuilder::NodeOut serialized; in Vectorize()
35 std::vector<NodeBuilder::NodeOut> dense_defaults; in Vectorize()
52 std::vector<NodeBuilder::NodeOut>* result) { in Vectorize()
63 std::vector<NodeBuilder::NodeOut> sparse_keys; in Vectorize()
66 std::vector<NodeBuilder::NodeOut> dense_keys; in Vectorize()
73 NodeBuilder(strings::StrCat("vectorized/", node.name()), "ParseExample") in Vectorize()
Ddecode_csv_vectorizer.cc30 NodeBuilder::NodeOut records; in Vectorize()
33 std::vector<NodeBuilder::NodeOut> defaults; in Vectorize()
40 auto node_builder = NodeBuilder(node.type_string(), node.type_string()) in Vectorize()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc39 return NodeBuilder(strings::StrCat(shared_name, suffix), op_name) in ReplaceReduce()
51 std::vector<NodeBuilder::NodeOut> out_nodes; in ReplaceReduce()
56 NodeBuilder recv_builder = in ReplaceReduce()
125 std::vector<std::forward_list<NodeBuilder::NodeOut>> out_nodes_map; in ReplaceBroadcast()
139 it->emplace_front(NodeBuilder::NodeOut(edge->dst(), edge->dst_input())); in ReplaceBroadcast()
163 return NodeBuilder(strings::StrCat(shared_name, suffix), op_name) in ReplaceBroadcast()
170 NodeBuilder::NodeOut in_node; in ReplaceBroadcast()
171 NodeBuilder send_builder = make_builder("_NcclBroadcastSend", "Send"); in ReplaceBroadcast()
176 in_node = NodeBuilder::NodeOut(edge->src(), edge->src_output()); in ReplaceBroadcast()
208 NodeBuilder shape_builder(shape_name, "Shape"); in ReplaceBroadcast()
[all …]
/external/llvm-project/polly/lib/CodeGen/
DCodeGeneration.cpp228 IslNodeBuilder NodeBuilder(Builder, Annotator, DL, LI, SE, DT, S, StartBlock); in CodeGen() local
232 NodeBuilder.allocateNewArrays(StartExitBlocks); in CodeGen()
250 if (!NodeBuilder.preloadInvariantLoads()) { in CodeGen()
272 NodeBuilder.addParameters(S.getContext().release()); in CodeGen()
273 Value *RTC = NodeBuilder.createRTC(AI.getRunCondition()); in CodeGen()
285 NodeBuilder.create(AstRoot); in CodeGen()
286 NodeBuilder.finalize(); in CodeGen()
296 for (auto *SubF : NodeBuilder.getParallelSubfunctions()) in CodeGen()
/external/webrtc/test/peer_scenario/tests/
Dremote_estimate_test.cc49 auto send_link = {s.net()->NodeBuilder().Build().node}; in TEST()
50 auto ret_link = {s.net()->NodeBuilder().Build().node}; in TEST()
81 auto send_node = s.net()->NodeBuilder().Build().node; in TEST()
82 auto ret_node = s.net()->NodeBuilder().Build().node; in TEST()

1234567