Searched refs:output_non_degenerated (Results 1 – 2 of 2) sorted by relevance
965 std::vector<int64> output_non_degenerated; in HandleReshape() local968 output_non_degenerated.push_back(i); in HandleReshape()971 if (output_non_degenerated.size() == 1) { in HandleReshape()972 output_dynamic_dimension = output_non_degenerated[0]; in HandleReshape()
3079 std::vector<int64> output_non_degenerated; in InferReshapeShape() local3082 output_non_degenerated.push_back(i); in InferReshapeShape()3085 if (output_non_degenerated.size() == 1) { in InferReshapeShape()3086 inferred_shape.set_dynamic_dimension(output_non_degenerated[0], true); in InferReshapeShape()