Searched refs:var_inputs (Results 1 – 1 of 1) sorted by relevance
51 std::vector<const Edge*> var_inputs; member195 (*execute_node_info)[j].var_inputs.push_back(edges[j]); in ExtractExecuteNodeInfo()208 TF_RET_CHECK(total == (*execute_node_info)[0].var_inputs.size()) in ExtractExecuteNodeInfo()210 << (*execute_node_info)[0].var_inputs.size(); in ExtractExecuteNodeInfo()536 AddNodeAttr("N", static_cast<int>(info.var_inputs.size()), in AddReshardOp()549 for (int i = 0; i < info.var_inputs.size(); ++i) { in AddReshardOp()550 const auto variable_edge = info.var_inputs[i]; in AddReshardOp()555 const int new_key_input = info.var_inputs.size(); in AddReshardOp()590 AddNodeAttr("N", static_cast<int>(info.var_inputs.size()), in AddReshardOp()598 for (int i = 0; i < info.var_inputs.size(); ++i) { in AddReshardOp()[all …]