Searched refs:output_dim_end (Results 1 – 2 of 2) sorted by relevance
835 int64 output_dim_end = -1; in HandleReshape() local846 output_dim_end = end.second; in HandleReshape()853 << " output dim end: " << output_dim_end; in HandleReshape()856 (output_dim_end - output_dim_start) > 1) { in HandleReshape()885 output_dim_end = std::max(output_dim_end, common_factor.second); in HandleReshape()910 output_dim_end - output_dim_start == 1) { 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()
2989 int64 output_dim_end = -1; in InferReshapeShape() local2998 output_dim_end = end.second; in InferReshapeShape()3003 (output_dim_end - output_dim_start) > 1) { in InferReshapeShape()3039 output_dim_end = std::max(output_dim_end, common_factor.second); in InferReshapeShape()3065 output_dim_end - output_dim_start == 1) { 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()