Home
last modified time | relevance | path

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

/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/core/common_runtime/
Dpartitioning_utils.cc78 std::vector<std::pair<Node*, int>> ret_nodes; in UpdateArgAndRetvalMetadata() local
94 ret_nodes.push_back(std::make_pair(node, index)); in UpdateArgAndRetvalMetadata()
109 for (int i = 0; i < ret_nodes.size(); ++i) { in UpdateArgAndRetvalMetadata()
110 Node* ret = ret_nodes[i].first; in UpdateArgAndRetvalMetadata()
Dfunction.cc1477 num_node_outputs != fbody->ret_nodes.size()) { in ValidateInlining()
1481 " ret_nodes=", fbody->ret_nodes.size()); in ValidateInlining()
1738 for (std::size_t i = 0; i < fbody->ret_nodes.size(); ++i) { in InlineFunctionBody()
1739 Node* ret = node_map[fbody->ret_nodes[i]->id()]; in InlineFunctionBody()
1927 this->ret_nodes.resize(ret_types.size()); in FunctionBody()
1931 node_vec = &this->ret_nodes; in FunctionBody()
2005 CHECK_EQ(fbody_->ret_types.size(), fbody_->ret_nodes.size()); in Copy()
2007 for (std::size_t i = 0; i < fbody_->ret_nodes.size(); ++i) { in Copy()
2008 gbody_->ret_nodes.push_back(node_map[fbody_->ret_nodes[i]->id()]); in Copy()
2021 const int num_y = static_cast<int>(gbody_->ret_nodes.size()); in Compute()
[all …]
Dfunction.h81 gtl::InlinedVector<Node*, 4> ret_nodes; member
Dprocess_function_library_runtime.cc499 ret_node_names->reserve(fbody->ret_nodes.size()); in GetGraphAndRets()
500 for (const Node* node : fbody->ret_nodes) { in GetGraphAndRets()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc48 indices->resize(fn_body->ret_nodes.size()); in ComputeShortCircuitIndices()
50 for (size_t i = 0; i < fn_body->ret_nodes.size(); ++i) { in ComputeShortCircuitIndices()
51 Node* ret_node = fn_body->ret_nodes[i]; in ComputeShortCircuitIndices()
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc176 Status GetRetDataTypes(const std::vector<Node*>& ret_nodes, in GetRetDataTypes() argument
178 send_from_host_dtypes->resize(ret_nodes.size(), DT_INVALID); in GetRetDataTypes()
179 for (auto* n : ret_nodes) { in GetRetDataTypes()
197 const std::vector<Node*>& ret_nodes, in BuildSendFromHostNode() argument
213 for (auto* n : ret_nodes) { in BuildSendFromHostNode()
240 std::vector<Node*> ret_nodes = GatherNodesWithType(*g, "_Retval"); in ReplaceRetNodesWithSendFromHostNode() local
241 TF_RETURN_IF_ERROR(GetRetDataTypes(ret_nodes, send_from_host_dtypes)); in ReplaceRetNodesWithSendFromHostNode()
244 BuildSendFromHostNode(g, oc_cluster_name, ret_nodes, in ReplaceRetNodesWithSendFromHostNode()
246 for (auto* n : ret_nodes) { in ReplaceRetNodesWithSendFromHostNode()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis.cc174 Node* ret_i = fbody->ret_nodes[i]; in GetCompileTimeConstInputs()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dwhile_op.cc110 for (int i = 0; i < body->ret_nodes.size(); i++) { in GetLoopInvariants()
112 const Node* ret = body->ret_nodes[i]; in GetLoopInvariants()