Home
last modified time | relevance | path

Searched refs:input_groups (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dinspecting_placer.cc37 for (int arg_index = 0; arg_index < input_groups.size(); ++arg_index) { in DebugString()
38 int group_id = input_groups[arg_index]; in DebugString()
154 converter.AssignGroups(fbody->arg_nodes, &groups->input_groups); in ComputeIOColocationGroups()
Dinspecting_placer.h55 std::vector<int> input_groups; member
Dcolocation_graph.cc906 for (int arg_idx = 0; arg_idx < groups.input_groups.size(); ++arg_idx) { in GetGroupNodes()
909 int group_id = groups.input_groups[arg_idx]; in GetGroupNodes()
946 if (groups.input_groups.size() != node.num_inputs()) { in ApplyIOColocationGroups()
950 groups.input_groups.size(), in ApplyIOColocationGroups()