Searched refs:num_reduced_args (Results 1 – 1 of 1) sorted by relevance
2125 int64 num_reduced_args = arg_shapes.size() / 2; in InferReduceShape() local2126 auto reduced_args = arg_shapes.subspan(0, num_reduced_args); in InferReduceShape()2129 for (int64 i = 1; i < num_reduced_args; ++i) { in InferReduceShape()2149 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape()2155 num_reduced_args)); in InferReduceShape()