| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gradients.cc | 43 string NodeOut::name() const { in name()
51 DataType NodeOut::dtype() const { return node->output_type(index); } in dtype()
54 uint64 operator()(const NodeOut& x) const { in operator ()()
61 bool operator()(const NodeOut& x, const NodeOut& y) const { in operator ()()
66 static Node* AddZerosLike(Graph* g, NodeOut input) { in AddZerosLike()
102 static Node* AddSymGrad(Graph* g, Node* n, gtl::ArraySlice<NodeOut> grads) { in AddSymGrad()
112 std::vector<NodeOut> n_inputs(num_x); in AddSymGrad()
118 for (const NodeOut& nout : n_inputs) { in AddSymGrad()
122 for (const NodeOut& nout : grads) { in AddSymGrad()
154 SymbolicGradientBuilder(gtl::ArraySlice<NodeOut> y_node_outputs,
[all …]
|
| 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 | lower_while_op.cc | 28 using NodeOut = NodeBuilder::NodeOut; typedef
244 .Input(NodeOut(edge->src(), edge->src_output())) in CreateEnterNodes()
286 .Input({NodeOut(enter_node, 0), NodeOut(enter_node, 0)}) in CreateMergeNodes()
298 cond_call_builder_.Input(NodeOut(enter_nodes_[i], 0)); in CreateCondFuncCallNode()
301 NodeOut(merge_nodes_[op_input_output_to_lowered_node_[i]], 0)); in CreateCondFuncCallNode()
312 .Input(NodeOut(cond_call_node_, 0)) in CreateCondFuncCallNode()
337 .Input(NodeOut(merge_node, 0)) in CreateSwitchNodes()
338 .Input(NodeOut(loop_cond_node_, 0)) in CreateSwitchNodes()
350 body_call_builder_.Input(NodeOut(enter_nodes_[i], 0)); in CreateBodyFuncCallNode()
353 NodeOut(switch_nodes_[op_input_output_to_lowered_node_[i]], 1)); in CreateBodyFuncCallNode()
[all …]
|
| D | lower_case_op.cc | 28 using NodeOut = NodeBuilder::NodeOut; typedef
67 std::vector<NodeOut> outputs_;
124 .Input(NodeOut(branch_index_)) in CreatePivotNodes()
125 .Input(NodeOut(branch_index_)) in CreatePivotNodes()
204 std::vector<NodeOut> merge_input; in AddOutputs()
214 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs()
227 std::vector<NodeOut> pivots(num_branches_); in AddOutputs()
229 pivots[j] = NodeOut(pivots_[j]); in AddOutputs()
|
| D | lower_if_op.cc | 26 using NodeOut = NodeBuilder::NodeOut; typedef
74 std::vector<NodeOut> outputs_;
155 .Input(NodeOut(pred_)) in CreatePivotNodes()
156 .Input(NodeOut(pred_)) in CreatePivotNodes()
242 .Input({NodeOut(then_call_node_, i), NodeOut(else_call_node_, i)}) in AddOutputs()
245 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs()
|
| D | lower_functional_ops_test.cc | 154 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()
261 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
|
| D | lower_while_op_test.cc | 69 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
191 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(placeholder)}); in TEST()
296 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(gpu_0_ph)}); in TEST()
419 std::vector<NodeBuilder::NodeOut> inputs( in TEST()
420 {NodeBuilder::NodeOut(a.node()), NodeBuilder::NodeOut(b.node())}); in TEST()
534 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
|
| D | lower_if_op_test.cc | 78 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
197 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(var.node())}); in TEST()
281 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
|
| D | lower_function_call_op_test.cc | 94 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
157 std::vector<NodeBuilder::NodeOut> inputs({NodeBuilder::NodeOut(a.node())}); in TEST()
|
| /external/tensorflow/tensorflow/core/graph/ |
| 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);
151 std::vector<NodeOut> inputs_;
|
| 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/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()
|
| /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/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 399 Node* RecvAtHost(ops::NodeOut key_input, const string& cluster, in RecvAtHost()
419 Node* SendFromHost(ops::NodeOut key_input, const string& cluster, in SendFromHost()
421 const std::vector<ops::NodeOut>& inputs, in SendFromHost()
444 Node* Unary(ops::NodeOut a, const GraphDefBuilder::Options& opts) { in Unary()
448 Node* Binary(ops::NodeOut a, ops::NodeOut b, in Binary()
453 Node* BinaryUnknownShape(ops::NodeOut a, ops::NodeOut b, in BinaryUnknownShape()
458 Node* AddNLike(const std::vector<ops::NodeOut>& inputs, in AddNLike()
472 Node* RetOp(int index, ops::NodeOut a, const GraphDefBuilder::Options& opts) { in RetOp()
902 ops::NodeOut(key_constant, 0), "F1", "F1", "O1", {DT_FLOAT, DT_FLOAT}, in TEST()
904 Node* e = Binary(ops::NodeOut(recv, 0), ops::NodeOut(recv, 1), in TEST()
[all …]
|
| /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()
158 using NodeOutList = std::vector<NodeDefBuilder::NodeOut>; in TEST()
161 NodeDefBuilder::NodeOut node_in{"a", 0, DT_STRING}; in TEST()
240 using NodeOutList = std::vector<NodeDefBuilder::NodeOut>; in TEST()
242 NodeDefBuilder::NodeOut string_in{"a", 0, DT_STRING}; in TEST()
308 using NodeOutList = std::vector<NodeDefBuilder::NodeOut>; in TEST()
310 NodeDefBuilder::NodeOut string_in{"a", 0, DT_STRING}; in TEST()
396 using NodeOutList = std::vector<NodeDefBuilder::NodeOut>; in TEST()
[all …]
|
| /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/framework/ |
| D | node_def_builder.h | 55 struct NodeOut { struct
56 NodeOut(StringPiece n, int i, DataType dt);
57 NodeOut(); // uninitialized, call Reset() before use.
81 NodeDefBuilder& Input(const NodeOut& src); argument
84 NodeDefBuilder& Input(gtl::ArraySlice<NodeOut> src_list);
162 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/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx_test.cc | 110 std::vector<NodeDefBuilder::NodeOut> src_list; in TEST_F()
125 std::vector<NodeDefBuilder::NodeOut> src_list; in TEST_F()
144 std::vector<NodeDefBuilder::NodeOut> src_list; in TEST_F()
|
| /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()
|