graph_partition.cc | 106 std::vector<DeviceType> device_types; member in struct:tensorflow::__anon39621::GraphInfo 131 if (info.device_types[src->id()] != DEVICE_CPU) { 146 if (info.device_types[dst->id()] != DEVICE_CPU) { 537 info->device_types.resize(g.num_node_ids(), DEVICE_CPU); 551 info->device_types[node_id] = DeviceType(parsed.type); [all...] |