Searched refs:node_shapes (Results 1 – 1 of 1) sorted by relevance
118 std::unordered_map<string, TensorShape>* node_shapes) { in GetOutputShapes() argument128 (*node_shapes)[wanted_shape] = input_tensor.second.shape(); in GetOutputShapes()143 (*node_shapes)[wanted_shape_name] = found_shape; in GetOutputShapes()