Home
last modified time | relevance | path

Searched refs:ret_nodes (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dpartitioning_utils.cc82 std::vector<std::pair<Node*, int>> ret_nodes; in UpdateArgAndRetvalMetadata() local
99 ret_nodes.emplace_back(node, index); in UpdateArgAndRetvalMetadata()
116 std::sort(ret_nodes.begin(), ret_nodes.end(), ret_comparator); in UpdateArgAndRetvalMetadata()
120 ret_indices->reserve(ret_nodes.size()); in UpdateArgAndRetvalMetadata()
121 for (const auto& pair : ret_nodes) ret_indices->push_back(pair.second); in UpdateArgAndRetvalMetadata()
140 for (int i = 0; i < ret_nodes.size(); ++i) { in UpdateArgAndRetvalMetadata()
141 Node* ret = ret_nodes[i].first; in UpdateArgAndRetvalMetadata()
Dfunction_body.cc31 this->ret_nodes.resize(ret_types.size()); in FunctionBody()
36 node_vec = &this->ret_nodes; in FunctionBody()
Dfunction_body.h41 gtl::InlinedVector<Node*, 4> ret_nodes; member
Dprocess_function_library_runtime.cc425 const std::vector<Node*>& arg_nodes, const std::vector<Node*>& ret_nodes, in PinArgsAndRets() argument
439 for (Node* node : ret_nodes) { in PinArgsAndRets()
635 std::vector<Node*>* arg_nodes, std::vector<Node*>* ret_nodes, in GetGraphAndArgRets() argument
650 ret_nodes->reserve(fbody->ret_nodes.size()); in GetGraphAndArgRets()
651 std::copy(fbody->ret_nodes.begin(), fbody->ret_nodes.end(), in GetGraphAndArgRets()
652 std::back_inserter(*ret_nodes)); in GetGraphAndArgRets()
654 ret_node_names->reserve(fbody->ret_nodes.size()); in GetGraphAndArgRets()
655 for (const Node* node : fbody->ret_nodes) { in GetGraphAndArgRets()
712 std::vector<Node*> arg_nodes, ret_nodes; in InstantiateMultiDevice() local
718 function_name, attrs, fdef, lib_def, &graph, &arg_nodes, &ret_nodes, in InstantiateMultiDevice()
[all …]
Dfunction.cc1433 CHECK_EQ(fbody_->ret_types.size(), fbody_->ret_nodes.size()); in Copy()
1435 for (std::size_t i = 0; i < fbody_->ret_nodes.size(); ++i) { in Copy()
1436 gbody->ret_nodes.push_back(node_map[fbody_->ret_nodes[i]->id()]); in Copy()
1446 const int num_y = static_cast<int>(gbody->ret_nodes.size()); in Compute()
1457 Node* y = gbody->ret_nodes[i]; in Compute()
1485 for (Node* n : gbody->ret_nodes) { in Compute()
1495 gbody->ret_nodes.clear(); in Compute()
1502 gbody->ret_nodes.push_back(ret); in Compute()
Dinline_function_utils.cc365 num_node_outputs != fbody->ret_nodes.size()) { in ValidateInlining()
369 " ret_nodes=", fbody->ret_nodes.size()); in ValidateInlining()
733 for (std::size_t i = 0; i < fbody->ret_nodes.size(); ++i) { in InlineFunctionBody()
734 Node* ret = node_map[fbody->ret_nodes[i]->id()]; in InlineFunctionBody()
Dinspecting_placer.cc155 converter.AssignGroups(fbody->ret_nodes, &groups->output_groups); in ComputeIOColocationGroups()
Dprocess_function_library_runtime.h388 const std::vector<Node*>& ret_nodes,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils.cc81 int index = map_defun_fn->ret_nodes.size(); in AddMapDefunOutput()
94 map_defun_fn->ret_nodes.push_back(ret_node); in AddMapDefunOutput()
103 DCHECK_LT(output_position, map_defun_fn->ret_nodes.size()) in RemoveMapDefunOutput()
107 int num_later_outputs = map_defun_fn->ret_nodes.size() - output_position - 1; in RemoveMapDefunOutput()
110 map_defun_fn->graph->RemoveNode(map_defun_fn->ret_nodes[output_position]); in RemoveMapDefunOutput()
111 map_defun_fn->ret_nodes.erase(map_defun_fn->ret_nodes.begin() + in RemoveMapDefunOutput()
122 map_defun_fn->ret_nodes[output_position + i]->AddAttr("index", in RemoveMapDefunOutput()
270 int output_index = map_defun_fn_->ret_nodes.size() - 1; in AddConversionMapping()
302 map_defun_fn_->ret_nodes[output_position]->input_edge(0, &ret_edge)); in ConvertOutput()
346 map_defun_fn_->ret_nodes); in VectorizeHelper()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Drearrange_function_argument.cc180 const gtl::InlinedVector<Node*, 4>& ret_nodes, // non-absl ok in CalculateRetvalRearrange() argument
183 for (int i = 0, end = ret_nodes.size(); i < end; i++) { in CalculateRetvalRearrange()
184 Node* n = ret_nodes[i]; in CalculateRetvalRearrange()
262 const gtl::InlinedVector<Node*, 4>& ret_nodes, // non-absl ok in RearrangeRetvalNodes() argument
264 for (int i = 0, end = ret_nodes.size(); i < end; i++) { in RearrangeRetvalNodes()
265 Node* n = ret_nodes[i]; in RearrangeRetvalNodes()
320 for (int i = 0, end = fbody->ret_nodes.size(); i < end; i++) { in MaybeRewriteWhileNode()
321 Node* n = fbody->ret_nodes[i]; in MaybeRewriteWhileNode()
352 for (int i = 0, end = fbody->ret_nodes.size(); i < end; i++) { in MaybeRewriteWhileNode()
353 Node* n = fbody->ret_nodes[i]; in MaybeRewriteWhileNode()
[all …]
Dconst_analysis.cc116 Node* ret_i = fbody->ret_nodes[i]; in GetCompileTimeConstInputs()
/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils.cc478 for (const auto& ret_node : fbody->ret_nodes) { in HashFunction()
534 if (this_fbody->ret_nodes.size() != that_fbody->ret_nodes.size()) { in CheckFunctionsEqualHelper()
537 that_name, ": ", this_fbody->ret_nodes.size(), " vs ", in CheckFunctionsEqualHelper()
538 that_fbody->ret_nodes.size()); in CheckFunctionsEqualHelper()
540 for (int i = 0; i < this_fbody->ret_nodes.size(); ++i) { in CheckFunctionsEqualHelper()
541 const NodeDef* this_root = &this_fbody->ret_nodes[i]->def(); in CheckFunctionsEqualHelper()
542 const NodeDef* that_root = &that_fbody->ret_nodes[i]->def(); in CheckFunctionsEqualHelper()
Dcaptured_function.cc191 indices.resize(fn_body->ret_nodes.size()); in CreateShortCircuitInfo()
193 for (size_t i = 0; i < fn_body->ret_nodes.size(); ++i) { in CreateShortCircuitInfo()
194 Node* ret_node = fn_body->ret_nodes[i]; in CreateShortCircuitInfo()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc219 absl::InlinedVector<OutputTensor, 4>* ret_nodes,
231 const absl::InlinedVector<OutputTensor, 4>& ret_nodes,
352 const absl::InlinedVector<OutputTensor, 4>& ret_nodes,
1216 absl::InlinedVector<OutputTensor, 4>* ret_nodes, in GetArgsAndRetsFromFunctionBody() argument
1219 ret_nodes->reserve(fbody.ret_nodes.size()); in GetArgsAndRetsFromFunctionBody()
1223 for (auto ret : fbody.ret_nodes) { in GetArgsAndRetsFromFunctionBody()
1224 ret_nodes->emplace_back(ret, 0); in GetArgsAndRetsFromFunctionBody()
1318 absl::InlinedVector<OutputTensor, 4> ret_nodes; in ConvertLibFunction() local
1320 GetArgsAndRetsFromFunctionBody(*fbody, &arg_nodes, &ret_nodes, in ConvertLibFunction()
1324 mlir_func_name, func_type, arg_nodes, ret_nodes, control_ret_nodes, in ConvertLibFunction()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.h66 std::vector<Node*>* arg_nodes, std::vector<Node*>* ret_nodes,
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc189 Status GetRetDataTypes(const std::vector<Node*>& ret_nodes, in GetRetDataTypes() argument
191 send_from_host_dtypes->resize(ret_nodes.size(), DT_INVALID); in GetRetDataTypes()
192 for (auto* n : ret_nodes) { in GetRetDataTypes()
210 const std::vector<Node*>& ret_nodes, in BuildSendFromHostNode() argument
226 for (auto* n : ret_nodes) { in BuildSendFromHostNode()
254 std::vector<Node*> ret_nodes = GatherNodesWithType(*g, "_Retval"); in ReplaceRetNodesWithSendFromHostNode() local
255 TF_RETURN_IF_ERROR(GetRetDataTypes(ret_nodes, send_from_host_dtypes)); in ReplaceRetNodesWithSendFromHostNode()
258 BuildSendFromHostNode(g, oc_cluster_name, ret_nodes, in ReplaceRetNodesWithSendFromHostNode()
260 for (auto* n : ret_nodes) { in ReplaceRetNodesWithSendFromHostNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc903 control_sources = {fbody.ret_nodes.begin(), fbody.ret_nodes.end()}; in ValidateSideEffectsExecution()
940 absl::flat_hash_set<const Node*> output_nodes = {fbody.ret_nodes.begin(), in ValidateNoDeadOutputs()
941 fbody.ret_nodes.end()}; in ValidateNoDeadOutputs()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc122 for (int i = 0; i < body->ret_nodes.size(); i++) { in GetLoopInvariants()
124 const Node* ret = body->ret_nodes[i]; in GetLoopInvariants()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc275 const Node* node = fbody->ret_nodes[out_id]; in MakeGrapplerFunctionItem()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1806 for (int i = 0; i < body_fbody->ret_nodes.size(); i++) { in FindArgsToLiftForWhileNode()
1807 const Node* node = body_fbody->ret_nodes[i]; in FindArgsToLiftForWhileNode()
2109 for (int i = 0; i < fbody->ret_nodes.size(); i++) { in CleanUpRetvalsForWhileBody()
2110 Node* ret_node = fbody->ret_nodes[i]; in CleanUpRetvalsForWhileBody()