Searched refs:execute_node_info (Results 1 – 1 of 1) sorted by relevance
115 std::vector<ExecuteNodeInfo>* execute_node_info, in ExtractExecuteNodeInfo() argument126 execute_node_info->clear(); in ExtractExecuteNodeInfo()129 execute_node_info->push_back({node}); in ExtractExecuteNodeInfo()132 if (execute_node_info->empty()) { in ExtractExecuteNodeInfo()135 TF_RET_CHECK(execute_node_info->size() == new_metadata->num_replicas()) in ExtractExecuteNodeInfo()137 << new_metadata->num_replicas() << " vs " << execute_node_info->size(); in ExtractExecuteNodeInfo()139 TF_RETURN_IF_ERROR(GetNodeAttr((*execute_node_info)[0].execute_node->attrs(), in ExtractExecuteNodeInfo()150 std::vector<const Edge*> edges(execute_node_info->size()); in ExtractExecuteNodeInfo()155 auto execute = (*execute_node_info)[j].execute_node; in ExtractExecuteNodeInfo()195 (*execute_node_info)[j].var_inputs.push_back(edges[j]); in ExtractExecuteNodeInfo()[all …]