| /external/tensorflow/tensorflow/core/graph/ |
| D | gradients.cc | 42 string NodeOut::name() const { in name()
50 DataType NodeOut::dtype() const { return node->output_type(index); } in dtype()
53 uint64 operator()(const NodeOut& x) const { in operator ()()
60 bool operator()(const NodeOut& x, const NodeOut& y) const { in operator ()()
65 static Node* AddZerosLike(Graph* g, NodeOut input) { in AddZerosLike()
101 static Node* AddSymGrad(Graph* g, Node* n, gtl::ArraySlice<NodeOut> grads) { in AddSymGrad()
111 std::vector<NodeOut> n_inputs(num_x); in AddSymGrad()
117 for (const NodeOut& nout : n_inputs) { in AddSymGrad()
121 for (const NodeOut& nout : grads) { in AddSymGrad()
153 SymbolicGradientBuilder(gtl::ArraySlice<NodeOut> y_node_outputs,
[all …]
|
| D | node_builder.cc | 25 NodeBuilder::NodeOut::NodeOut(Node* n, int32 i) // NOLINT(runtime/explicit) in NodeOut() function in tensorflow::NodeBuilder::NodeOut
32 NodeBuilder::NodeOut::NodeOut(OutputTensor t) : NodeOut(t.node, t.index) {} in NodeOut() function in tensorflow::NodeBuilder::NodeOut
34 NodeBuilder::NodeOut::NodeOut(StringPiece n, int32 i, DataType t) in NodeOut() function in tensorflow::NodeBuilder::NodeOut
37 NodeBuilder::NodeOut::NodeOut() in NodeOut() function in tensorflow::NodeBuilder::NodeOut
60 NodeBuilder& NodeBuilder::Input(NodeOut src) { in Input()
70 NodeBuilder& NodeBuilder::Input(gtl::ArraySlice<NodeOut> src_list) { in Input()
71 std::vector<NodeDefBuilder::NodeOut> srcs; in Input()
81 def_builder_.Input(gtl::ArraySlice<NodeDefBuilder::NodeOut>(srcs)); in Input()
|
| D | node_builder.h | 50 struct NodeOut { struct
52 NodeOut(Node* n, int32 i = 0);
53 NodeOut(OutputTensor t);
59 NodeOut(StringPiece name, int32 i, DataType t);
62 NodeOut();
92 NodeBuilder& Input(NodeOut src); argument
95 NodeBuilder& Input(gtl::ArraySlice<NodeOut> src_list);
149 std::vector<NodeOut> inputs_;
|
| D | gradients.h | 26 struct NodeOut { struct
50 Status AddSymbolicGradients(gtl::ArraySlice<NodeOut> y_node_outputs, argument
51 gtl::ArraySlice<NodeOut> x_node_outputs,
52 gtl::ArraySlice<NodeOut> y_grad_node_outputs,
53 std::vector<NodeOut>* x_grad_node_outputs,
|
| 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 | graph_def_builder.h | 190 typedef NodeBuilder::NodeOut NodeOut; typedef
196 Node* UnaryOp(const string& op_name, NodeOut input,
200 Node* BinaryOp(const string& op_name, NodeOut a, NodeOut b,
|
| D | graph_def_builder.cc | 108 Node* UnaryOp(const string& op_name, NodeOut input, in UnaryOp()
117 Node* BinaryOp(const string& op_name, NodeOut a, NodeOut b, in BinaryOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 411 Node* RecvAtHost(ops::NodeOut key_input, const string& cluster, in RecvAtHost()
429 Node* SendFromHost(ops::NodeOut key_input, const string& cluster, in SendFromHost()
431 const std::vector<ops::NodeOut>& inputs, in SendFromHost()
453 Node* Unary(ops::NodeOut a, const GraphDefBuilder::Options& opts) { in Unary()
457 Node* Binary(ops::NodeOut a, ops::NodeOut b, in Binary()
462 Node* BinaryUnknownShape(ops::NodeOut a, ops::NodeOut b, in BinaryUnknownShape()
467 Node* AddNLike(const std::vector<ops::NodeOut>& inputs, in AddNLike()
481 Node* RetOp(int index, ops::NodeOut a, const GraphDefBuilder::Options& opts) { in RetOp()
909 ops::NodeOut(key_constant, 0), "F1", "O1", {DT_FLOAT, DT_FLOAT}, in TEST()
911 Node* e = Binary(ops::NodeOut(recv, 0), ops::NodeOut(recv, 1), in TEST()
[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/cc/ops/ |
| D | const_op.cc | 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);
77 std::vector<NodeBuilder::NodeOut> AsNodeOutList(const Scope& scope,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 28 using NodeOut = NodeBuilder::NodeOut; typedef
200 .Input(NodeOut(edge->src(), edge->src_output())) in CreateEnterNodes()
235 .Input({NodeOut(enter_node, 0), NodeOut(enter_node, 0)}) in CreateMergeNodes()
245 cond_call_builder_.Input(NodeOut(merge_node, 0)); in CreateCondFuncCallNode()
255 .Input(NodeOut(cond_call_node_, 0)) in CreateCondFuncCallNode()
276 .Input(NodeOut(merge_nodes_[i], 0)) in CreateSwitchNodes()
277 .Input(NodeOut(loop_cond_node_, 0)) in CreateSwitchNodes()
287 body_call_builder_.Input(NodeOut(switch_node, 1)); in CreateBodyFuncCallNode()
305 .Input(NodeOut(switch_nodes_[0], 1)) in CreateBodyFuncCallNode()
313 std::vector<NodeOut> outputs; in CreateExitNodes()
[all …]
|
| D | lower_if_op.cc | 27 using NodeOut = NodeBuilder::NodeOut; typedef
70 std::vector<NodeOut> outputs_;
125 .Input(NodeOut(pred_)) in CreatePivotNodes()
126 .Input(NodeOut(pred_)) in CreatePivotNodes()
203 .Input({NodeOut(then_call_node_, i), NodeOut(else_call_node_, i)}) in AddOutputs()
206 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs()
|
| D | lower_while_op_test.cc | 59 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
161 std::vector<NodeBuilder::NodeOut> inputs( in TEST()
162 {NodeBuilder::NodeOut(a.node()), NodeBuilder::NodeOut(b.node())}); in TEST()
264 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
|
| D | lower_if_while_test.cc | 142 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()
249 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
|
| D | lower_if_op_test.cc | 70 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
188 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(var.node())}); in TEST()
272 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); 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()
|
| D | decode_csv_vectorizer.cc | 30 NodeBuilder::NodeOut records; in Vectorize()
33 std::vector<NodeBuilder::NodeOut> defaults; in Vectorize()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | parsing_ops_test.cc | 36 std::vector<NodeDefBuilder::NodeOut> src_list; in TEST()
91 using NodeOutList = std::vector<NodeDefBuilder::NodeOut>; in TEST()
93 NodeDefBuilder::NodeOut string_in{"a", 0, DT_STRING}; in TEST()
155 using NodeOutList = std::vector<NodeDefBuilder::NodeOut>; in TEST()
158 NodeDefBuilder::NodeOut node_in{"a", 0, DT_STRING}; in TEST()
237 using NodeOutList = std::vector<NodeDefBuilder::NodeOut>; in TEST()
239 NodeDefBuilder::NodeOut string_in{"a", 0, DT_STRING}; in TEST()
|
| D | image_ops_test.cc | 303 .Input(NodeDefBuilder::NodeOut{"images", 0, DT_QINT32}) in TEST()
304 .Input(NodeDefBuilder::NodeOut{"size", 0, DT_INT32}) in TEST()
305 .Input(NodeDefBuilder::NodeOut{"min", 0, DT_FLOAT}) in TEST()
306 .Input(NodeDefBuilder::NodeOut{"max", 0, DT_FLOAT}) in TEST()
|
| D | control_flow_ops_test.cc | 29 std::vector<NodeDefBuilder::NodeOut> src_list; in TEST()
56 std::vector<NodeDefBuilder::NodeOut> src_list; in TEST()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_builder.h | 53 struct NodeOut { struct
54 NodeOut(StringPiece n, int i, DataType dt);
55 NodeOut(); // uninitialized, call Reset() before use.
79 NodeDefBuilder& Input(const NodeOut& src); argument
82 NodeDefBuilder& Input(gtl::ArraySlice<NodeOut> src_list);
156 gtl::ArraySlice<NodeOut> src_list);
|
| D | node_def_builder.cc | 27 NodeDefBuilder::NodeOut::NodeOut(StringPiece n, int i, DataType dt) in NodeOut() function in tensorflow::NodeDefBuilder::NodeOut
30 NodeDefBuilder::NodeOut::NodeOut() { in NodeOut() function in tensorflow::NodeDefBuilder::NodeOut
34 void NodeDefBuilder::NodeOut::Reset(StringPiece n, int i, DataType dt) { in Reset()
103 NodeDefBuilder& NodeDefBuilder::Input(const NodeOut& src) { in Input()
109 NodeDefBuilder& NodeDefBuilder::Input(gtl::ArraySlice<NodeOut> src_list) { in Input()
137 gtl::ArraySlice<NodeOut> src_list) { in ListInput()
|
| D | fake_input.cc | 171 std::vector<NodeDefBuilder::NodeOut> srcs; in NSources()
176 builder_->Input(gtl::ArraySlice<NodeDefBuilder::NodeOut>(srcs)); in NSources()
180 std::vector<NodeDefBuilder::NodeOut> srcs; in SourceList()
185 builder_->Input(gtl::ArraySlice<NodeDefBuilder::NodeOut>(srcs)); in SourceList()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 51 std::vector<NodeBuilder::NodeOut> out_nodes; 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()
170 NodeBuilder::NodeOut in_node; in ReplaceBroadcast()
176 in_node = NodeBuilder::NodeOut(edge->src(), edge->src_output()); in ReplaceBroadcast()
|