Searched refs:num_reduced_args (Results 1 – 1 of 1) sorted by relevance
2054 int64 num_reduced_args = arg_shapes.size() / 2; in InferReduceShape() local2056 auto reduced_args = arg_shapes.subspan(0, num_reduced_args); in InferReduceShape()2059 for (int64 i = 1; i < num_reduced_args; ++i) { in InferReduceShape()2080 auto init_values = arg_shapes.subspan(num_reduced_args, arg_shapes.size()); in InferReduceShape()2086 num_reduced_args)); in InferReduceShape()