Searched refs:output_dim_start (Results 1 – 2 of 2) sorted by relevance
834 int64 output_dim_start = -1; in HandleReshape() local845 output_dim_start = start.second; in HandleReshape()852 << " output dim start: " << output_dim_start in HandleReshape()856 (output_dim_end - output_dim_start) > 1) { in HandleReshape()882 output_dim_start = std::min(output_dim_start, common_factor.second); in HandleReshape()910 output_dim_end - output_dim_start == 1) { in HandleReshape()912 output_dynamic_dimension = output_dim_start; in HandleReshape()916 output_dim_end - output_dim_start > 1) { in HandleReshape()957 for (int64 i = output_dim_start; i < output_dim_end; ++i) { in HandleReshape()966 for (int64 i = output_dim_start; i < output_dim_end; ++i) { in HandleReshape()
2988 int64 output_dim_start = -1; in InferReshapeShape() local2997 output_dim_start = start.second; in InferReshapeShape()3003 (output_dim_end - output_dim_start) > 1) { in InferReshapeShape()3036 output_dim_start = std::min(output_dim_start, common_factor.second); in InferReshapeShape()3065 output_dim_end - output_dim_start == 1) { in InferReshapeShape()3067 output_dynamic_dimension = output_dim_start; in InferReshapeShape()3070 output_dim_end - output_dim_start > 1) { in InferReshapeShape()3080 for (int64 i = output_dim_start; i < output_dim_end; ++i) { in InferReshapeShape()