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