Searched refs:wanted_shape (Results 1 – 1 of 1) sorted by relevance
124 for (const string& wanted_shape : wanted_shapes) { in GetOutputShapes() local128 if (input_tensor.first == wanted_shape) { in GetOutputShapes()129 (*node_shapes)[wanted_shape] = input_tensor.second.shape(); in GetOutputShapes()135 output_tensor_names.push_back(wanted_shape); in GetOutputShapes()