Searched refs:reduced_args (Results 1 – 1 of 1) sorted by relevance
2056 auto reduced_args = arg_shapes.subspan(0, num_reduced_args); in InferReduceShape() local2060 if (!ShapeUtil::SameDimensions(*reduced_args[0], *reduced_args[i])) { in InferReduceShape()2064 ShapeUtil::HumanString(*reduced_args[0]), i, in InferReduceShape()2065 ShapeUtil::HumanString(*reduced_args[i])); in InferReduceShape()2072 const Shape& arg = *reduced_args[0]; in InferReduceShape()2082 for (const Shape* arg : reduced_args) { in InferReduceShape()