| /external/tensorflow/tensorflow/core/graph/ |
| D | node_builder.cc | 25 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 …]
|
| D | node_builder.h | 42 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 …]
|
| D | testlib.cc | 38 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 …]
|
| D | node_builder_test.cc | 33 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()
|
| D | quantize_training.cc | 203 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 …]
|
| D | subgraph.h | 59 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,
|
| D | mkl_layout_pass.cc | 828 std::function<void(const Node*, NodeBuilder*, bool)> copy_attrs;
868 std::function<void(const Node*, NodeBuilder* nb, bool)>)>
870 std::function<void(const Node*, NodeBuilder* nb, bool)> copy_attrs;
1325 std::function<void(const Node*, NodeBuilder* nb, bool)> copy_attrs,
1752 std::vector<NodeBuilder::NodeOut>* output_nodes);
1773 std::vector<NodeBuilder::NodeOut>* output_nodes);
1808 NodeBuilder* nb, const Node* old_node,
1809 std::vector<NodeBuilder::NodeOut>* workspace_tensors,
1822 NodeBuilder* nb, const Node* orig_node);
1833 NodeBuilder* nb);
[all …]
|
| D | subgraph.cc | 220 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()
|
| D | graph_def_builder.cc | 79 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/tensorflow/tensorflow/compiler/tf2xla/ |
| D | resource_util_test.cc | 95 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/cc/ops/ |
| D | const_op.cc | 30 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()
|
| D | const_op.h | 33 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,
|
| /external/clang/include/clang/StaticAnalyzer/Core/PathSensitive/ |
| D | CoreEngine.h | 32 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/ |
| D | sdca_ops_test.cc | 160 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/common_runtime/ |
| D | lower_if_op.cc | 27 using NodeOut = NodeBuilder::NodeOut;
101 NodeBuilder then_call_builder_;
102 NodeBuilder else_call_builder_;
134 TF_RETURN_IF_ERROR(NodeBuilder(NewName("switch_pred"), "Switch", in CreatePivotNodes()
141 TF_RETURN_IF_ERROR(NodeBuilder(NewName("pivot_f"), "Identity", in CreatePivotNodes()
146 TF_RETURN_IF_ERROR(NodeBuilder(NewName("pivot_t"), "Identity", in CreatePivotNodes()
170 TF_RETURN_IF_ERROR(NodeBuilder(NewName(src->name()), "Switch", in AddInput()
212 NodeBuilder(NewName("output"), "Merge", graph_->op_registry(), in AddOutputs()
230 TF_RETURN_IF_ERROR(NodeBuilder(NewName("branch_executed"), "Merge", in AddOutputs()
260 NodeBuilder builder = keep_node_fetchable_ && !outputs_.empty() in BuildLoweredIfOutput()
[all …]
|
| D | lower_while_op.cc | 30 using NodeOut = NodeBuilder::NodeOut;
157 NodeBuilder cond_call_builder_;
158 NodeBuilder body_call_builder_;
243 NodeBuilder builder = NodeBuilder(NewName("enter"), "Enter", in CreateEnterNodes()
265 TF_RETURN_IF_ERROR(NodeBuilder(NewName("LoopControlInputs"), "NoOp", in CreateEnterNodes()
284 NodeBuilder(NewName("merge"), "Merge", graph_->op_registry(), in CreateMergeNodes()
309 TF_RETURN_IF_ERROR(NodeBuilder(NewName("LoopCond"), "LoopCond", in CreateCondFuncCallNode()
336 NodeBuilder(NewName(op_name), op_type, graph_->op_registry(), in CreateSwitchNodes()
371 TF_RETURN_IF_ERROR(NodeBuilder(NewName("loop_body_control"), op_type, in CreateBodyFuncCallNode()
393 NodeBuilder(NewName("exit"), "Exit", graph_->op_registry(), in CreateExitNodes()
[all …]
|
| D | lower_case_op.cc | 29 using NodeOut = NodeBuilder::NodeOut;
96 std::vector<NodeBuilder> branch_call_builders_;
123 TF_RETURN_IF_ERROR(NodeBuilder(NewName("branch_index"), "_SwitchN", in CreatePivotNodes()
133 TF_RETURN_IF_ERROR(NodeBuilder(NewName(strings::StrCat("pivot_", b)), in CreatePivotNodes()
159 TF_RETURN_IF_ERROR(NodeBuilder(NewName(src->name()), "_SwitchN", in AddInput()
210 TF_RETURN_IF_ERROR(NodeBuilder(NewName("merge"), "Merge", in AddOutputs()
232 TF_RETURN_IF_ERROR(NodeBuilder(NewName("branch_executed"), "Merge", in AddOutputs()
261 NodeBuilder builder = keep_node_fetchable_ && !outputs_.empty() in BuildLoweredCaseOutput()
262 ? NodeBuilder(name_, "IdentityN").Input(outputs_) in BuildLoweredCaseOutput()
263 : NodeBuilder(name_, "NoOp"); in BuildLoweredCaseOutput()
|
| D | shape_refiner_test.cc | 80 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 …]
|
| D | lower_function_call_op_test.cc | 96 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
97 TF_ASSERT_OK(NodeBuilder("F", "PartitionedCall", &root.graph()->flib_def()) in TEST()
159 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
162 TF_ASSERT_OK(NodeBuilder("B", "PartitionedCall", &root.graph()->flib_def()) in TEST()
171 TF_ASSERT_OK(NodeBuilder("C", "PartitionedCall", &root.graph()->flib_def()) in TEST()
|
| D | lower_while_op_test.cc | 69 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
75 NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
187 std::vector<NodeBuilder::NodeOut> inputs( in TEST()
188 {NodeBuilder::NodeOut(a.node()), NodeBuilder::NodeOut(b.node())}); in TEST()
194 NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
302 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
308 NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
|
| D | lower_functional_ops_test.cc | 155 std::vector<NodeBuilder::NodeOut> inputs( in TEST()
156 {NodeBuilder::NodeOut(counter.node()), NodeBuilder::NodeOut(pred.node()), in TEST()
157 NodeBuilder::NodeOut(a.node())}); in TEST()
163 TF_ASSERT_OK(NodeBuilder("while", "While", &root.graph()->flib_def()) in TEST()
262 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
268 TF_ASSERT_OK(NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
|
| D | lower_if_op_test.cc | 80 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
82 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
199 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(var.node())}); in TEST()
201 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
283 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
289 NodeBuilder("if", "If", &root.graph()->flib_def()) in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| D | parse_single_example_vectorizer.cc | 32 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()
|
| D | decode_csv_vectorizer.cc | 30 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/ |
| D | nccl_rewrite.cc | 39 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 …]
|