Home
last modified time | relevance | path

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

12345

/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()
Dquantize_training.cc203 NodeBuilder save_op_builder = in ConnectVariablesToSaveOp()
204 NodeBuilder(save_op->name(), save_op->type_string()); in ConnectVariablesToSaveOp()
209 std::vector<NodeBuilder::NodeOut> var_nodeouts; in ConnectVariablesToSaveOp()
279 TF_RETURN_IF_ERROR(NodeBuilder(tensor_names_op_name, "Const") in AddRestoreVariableSubgraphs()
288 TF_RETURN_IF_ERROR(NodeBuilder(shape_and_slices_op_name, "Const") in AddRestoreVariableSubgraphs()
295 TF_RETURN_IF_ERROR(NodeBuilder(new_restore_op_name, "RestoreV2") in AddRestoreVariableSubgraphs()
304 TF_RETURN_IF_ERROR(NodeBuilder(new_assign_op_name, "Assign") in AddRestoreVariableSubgraphs()
340 NodeBuilder(strings::StrCat(name_prefix, "/RangeStart"), "Const") in MakeReductionAxes()
348 NodeBuilder(strings::StrCat(name_prefix, "/RangeDelta"), "Const") in MakeReductionAxes()
354 NodeBuilder(strings::StrCat(name_prefix, "/InputRank"), "Rank") in MakeReductionAxes()
[all …]
Dmkl_layout_pass.cc608 std::function<void(const Node*, NodeBuilder*, bool)> copy_attrs;
646 std::function<void(const Node*, NodeBuilder* nb, bool)>)>
648 std::function<void(const Node*, NodeBuilder* nb, bool)> copy_attrs;
1082 std::function<void(const Node*, NodeBuilder* nb, bool)> copy_attrs,
1361 std::vector<NodeBuilder::NodeOut>* output_nodes);
1382 std::vector<NodeBuilder::NodeOut>* output_nodes);
1416 NodeBuilder* nb, Node* old_node,
1417 std::vector<NodeBuilder::NodeOut>* workspace_tensors,
1430 NodeBuilder* nb, Node* orig_node);
1438 NodeBuilder* nb,
[all …]
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,
Dsubgraph.cc220 Status ArgFeedRewrite::AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor, in AddNode()
227 NodeBuilder(strings::StrCat("_arg_", feed_tensor.node->name(), "_", in AddNode()
237 Status RecvFeedRewrite::AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor, in AddNode()
240 NodeBuilder(strings::StrCat("_recv_", feed_tensor.node->name(), "_", in AddNode()
257 Status RetvalFetchRewrite::AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor, in AddNode()
264 NodeBuilder(strings::StrCat("_retval_", fetch_tensor.node->name(), "_", in AddNode()
276 Status SendFetchRewrite::AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor, in AddNode()
279 NodeBuilder(strings::StrCat("_send_", fetch_tensor.node->name(), "_", in AddNode()
Dgraph_def_builder.cc79 Node* GraphDefBuilder::Options::FinalizeBuilder(NodeBuilder* builder) const { in FinalizeBuilder()
103 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in SourceOp()
111 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in UnaryOp()
120 NodeBuilder node_builder(opts.GetNameForOp(op_name), op_name, in BinaryOp()
/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/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()
Dconst_op.h33 NodeBuilder::NodeOut AsNodeOut(const Scope& scope, const Input& inp);
55 auto cast_builder = NodeBuilder(cast_op_name, "Cast") in Const()
77 std::vector<NodeBuilder::NodeOut> AsNodeOutList(const Scope& scope,
Dwhile_loop.cc71 NodeBuilder::NodeOut enter_input(enter_output.node(), enter_output.index()); in CreateMerge()
75 NodeBuilder::NodeOut next_input(NextIterationName(scope, loop_var_idx), in CreateMerge()
78 std::vector<NodeBuilder::NodeOut> input_list({enter_input, next_input}); in CreateMerge()
80 NodeBuilder builder = NodeBuilder(unique_name, "Merge").Input(input_list); in CreateMerge()
/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/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/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/common_runtime/
Dlower_while_op.cc28 using NodeOut = NodeBuilder::NodeOut;
140 NodeBuilder cond_call_builder_;
141 NodeBuilder body_call_builder_;
198 TF_RETURN_IF_ERROR(NodeBuilder(NewName("enter"), "Enter", in CreateEnterNodes()
217 TF_RETURN_IF_ERROR(NodeBuilder(NewName("LoopControlInputs"), "NoOp", in CreateEnterNodes()
233 NodeBuilder(NewName("merge"), "Merge", graph_->op_registry(), in CreateMergeNodes()
253 TF_RETURN_IF_ERROR(NodeBuilder(NewName("LoopCond"), "LoopCond", in CreateCondFuncCallNode()
274 TF_RETURN_IF_ERROR(NodeBuilder(NewName(op_name), op_type, in CreateSwitchNodes()
303 TF_RETURN_IF_ERROR(NodeBuilder(NewName("loop_body_control"), op_type, in CreateBodyFuncCallNode()
317 TF_RETURN_IF_ERROR(NodeBuilder(NewName("exit"), "Exit", in CreateExitNodes()
[all …]
Dlower_if_op.cc27 using NodeOut = NodeBuilder::NodeOut;
98 NodeBuilder then_call_builder_;
99 NodeBuilder else_call_builder_;
123 TF_RETURN_IF_ERROR(NodeBuilder(NewName("switch_pred"), "Switch", in CreatePivotNodes()
130 TF_RETURN_IF_ERROR(NodeBuilder(NewName("pivot_f"), "Identity", in CreatePivotNodes()
135 TF_RETURN_IF_ERROR(NodeBuilder(NewName("pivot_t"), "Identity", in CreatePivotNodes()
159 TF_RETURN_IF_ERROR(NodeBuilder(NewName(src->name()), "Switch", in AddInput()
201 NodeBuilder(graph_->NewName("output"), "Merge", graph_->op_registry(), in AddOutputs()
216 TF_RETURN_IF_ERROR(NodeBuilder(graph_->NewName("branch_executed"), "Merge", in AddOutputs()
274 NodeBuilder ib(name_, outputs_.empty() ? "NoOp" : "IdentityN"); in BuildLoweredIfOutput()
Dshape_refiner_test.cc80 TF_ASSERT_OK(NodeBuilder("test", "TensorAsShapeInt32") in TestStridedSlice()
227 TF_ASSERT_OK(NodeBuilder("b", "TestOpWithNoShapeFn") in TEST_F()
301 TF_ASSERT_OK(NodeBuilder("MI", "MultiIdentity") in TEST_F()
302 .Input(std::vector<NodeBuilder::NodeOut>{small.node(), in TEST_F()
308 TF_ASSERT_OK(NodeBuilder("Test", "ShapeVectorForAllElements") in TEST_F()
314 TF_ASSERT_OK(NodeBuilder("Test", "ShapeVectorForAllElements") in TEST_F()
368 TF_ASSERT_OK(NodeBuilder("Test", "TestOp") in TEST_F()
483 TF_ASSERT_OK(NodeBuilder("Test", "ShapeData") in TEST_F()
517 TF_ASSERT_OK(NodeBuilder("Test", "ShapeDataInt64") in TEST_F()
550 TF_ASSERT_OK(NodeBuilder("Test", "ShapeData") in TEST_F()
[all …]
Dlower_while_op_test.cc59 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
64 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
161 std::vector<NodeBuilder::NodeOut> inputs( in TEST()
162 {NodeBuilder::NodeOut(a.node()), NodeBuilder::NodeOut(b.node())}); in TEST()
167 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
264 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
269 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
Dlower_if_while_test.cc142 std::vector<NodeBuilder::NodeOut> inputs( in TEST()
143 {NodeBuilder::NodeOut(counter.node()), NodeBuilder::NodeOut(pred.node()), in TEST()
144 NodeBuilder::NodeOut(a.node())}); in TEST()
150 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
249 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
255 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
323 TF_ASSERT_OK(NodeBuilder("const", "Const") in TEST()
Dlower_if_op_test.cc70 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
71 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
188 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(var.node())}); in TEST()
189 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
272 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
277 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
/external/tensorflow/tensorflow/lite/tools/accuracy/
Drun_tflite_model_op_test.cc50 std::vector<NodeBuilder::NodeOut> input_data; in TEST()
59 auto builder = ::tensorflow::NodeBuilder("run_model_op", "RunTFLiteModel") in TEST()
110 std::vector<NodeBuilder::NodeOut> input_data; in TEST()
119 auto builder = ::tensorflow::NodeBuilder("run_model_op", "RunTFLiteModel") in TEST()
157 std::vector<NodeBuilder::NodeOut> input_data; in TEST()
166 auto builder = ::tensorflow::NodeBuilder("run_model_op", "RunTFLiteModel") in TEST()
/external/clang/lib/StaticAnalyzer/Core/
DCheckerManager.cpp120 NodeBuilder B(*PrevSet, *CurrSet, BldrCtx); in expandGraphWithCheckers()
153 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
198 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
265 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
309 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
354 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
395 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
425 NodeBuilder Bldr(Pred, Dst, BC); in runCheckersForEndFunction()
452 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
497 NodeBuilder &Bldr, ExplodedNode *Pred) { in runChecker()
[all …]

12345