Searched refs:reduced_args (Results 1 – 1 of 1) sorted by relevance
2126 auto reduced_args = arg_shapes.subspan(0, num_reduced_args); in InferReduceShape() local2130 if (!ShapeUtil::SameDimensions(*reduced_args[0], *reduced_args[i])) { in InferReduceShape()2134 ShapeUtil::HumanString(*reduced_args[0]), i, in InferReduceShape()2135 ShapeUtil::HumanString(*reduced_args[i])); in InferReduceShape()2141 const Shape& arg = *reduced_args[0]; in InferReduceShape()2151 for (const Shape* arg : reduced_args) { in InferReduceShape()