HomeSort by relevance Sort by last modified time
    Searched defs:ret_nodes (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
function.h 81 gtl::InlinedVector<Node*, 4> ret_nodes; member in struct:tensorflow::FunctionBody
partitioning_utils.cc 78 std::vector<std::pair<Node*, int>> ret_nodes; local
94 ret_nodes.push_back(std::make_pair(node, index));
109 for (int i = 0; i < ret_nodes.size(); ++i) {
110 Node* ret = ret_nodes[i].first;
  /external/tensorflow/tensorflow/compiler/jit/
extract_outside_compilation_pass.cc 175 // Gets data types from `ret_nodes` and fills them into `send_from_host_dtypes`.
176 Status GetRetDataTypes(const std::vector<Node*>& ret_nodes,
178 send_from_host_dtypes->resize(ret_nodes.size(), DT_INVALID);
179 for (auto* n : ret_nodes) {
197 const std::vector<Node*>& ret_nodes,
213 for (auto* n : ret_nodes) {
240 std::vector<Node*> ret_nodes = GatherNodesWithType(*g, "_Retval"); local
241 TF_RETURN_IF_ERROR(GetRetDataTypes(ret_nodes, send_from_host_dtypes));
244 BuildSendFromHostNode(g, oc_cluster_name, ret_nodes,
246 for (auto* n : ret_nodes) {
    [all...]

Completed in 269 milliseconds