Searched refs:body_nodes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function.cc | 114 std::vector<const Node*>* body_nodes) in ComputeBodyNodes() argument
121 body_nodes->push_back(node); in ComputeBodyNodes()
136 body_nodes->reserve(num_opers); in ComputeBodyNodes()
139 body_nodes->push_back(node); in ComputeBodyNodes()
193 std::vector<const Node*> body_nodes; in TF_GraphToFunctionWithControlOutputs() local
195 fn_body, fn_name, num_opers, opers, input_nodes, &body_nodes); in TF_GraphToFunctionWithControlOutputs()
211 /*copy_placeholder_attrs_from_nodes=*/true, body_nodes, input_tensors, in TF_GraphToFunctionWithControlOutputs()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef.cc | 157 const std::vector<const Node*>& body_nodes, in FillFunctionBody() argument
168 for (const Node* node : body_nodes) { in FillFunctionBody()
315 std::vector<const Node*> body_nodes; in GraphToFunctionDefHelper() local
339 body_nodes.push_back(node); in GraphToFunctionDefHelper()
360 body_nodes, inputs, outputs, output_names, in GraphToFunctionDefHelper()
371 const std::vector<const Node*>& body_nodes, in GraphToFunctionDef() argument
476 for (const Node* node : body_nodes) { in GraphToFunctionDef()
502 TF_RETURN_IF_ERROR(FillFunctionBody(fn_name, node_names, body_nodes, in GraphToFunctionDef()
|
| D | graph_to_functiondef.h | 35 const std::vector<const Node*>& body_nodes,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor_test.cc | 618 std::vector<FunctionDefHelper::Node> body_nodes; in BM_WhileLoopHelper() local
619 body_nodes.reserve(1 + loop_vars); in BM_WhileLoopHelper()
620 body_nodes.push_back( in BM_WhileLoopHelper()
622 body_nodes.push_back({{"y"}, "Add", {"x", "one"}, {{"T", DT_INT32}}}); in BM_WhileLoopHelper()
624 body_nodes.push_back({{strings::StrCat("y", i)}, in BM_WhileLoopHelper()
640 body_nodes); in BM_WhileLoopHelper()
|