Searched refs:unmodified_dims (Results 1 – 2 of 2) sorted by relevance
1057 std::vector<std::pair<int64, int64>> unmodified_dims = in ReshapeLeavesDimensionsUnmodified() local1064 while (i < unmodified_dims.size() && in ReshapeLeavesDimensionsUnmodified()1065 unmodified_dims[i].first < input_dim_index) { in ReshapeLeavesDimensionsUnmodified()1068 if (i >= unmodified_dims.size() || in ReshapeLeavesDimensionsUnmodified()1069 unmodified_dims[i].first != input_dim_index) { in ReshapeLeavesDimensionsUnmodified()1072 output_dim_indices.push_back(unmodified_dims[i].second); in ReshapeLeavesDimensionsUnmodified()1642 std::vector<std::pair<int64, int64>> unmodified_dims = in HandleReduce() local1650 for (auto dim_pair : unmodified_dims) { in HandleReduce()1665 for (auto dim_pair : unmodified_dims) { in HandleReduce()
1031 std::vector<std::pair<int64, int64>> unmodified_dims = in InsertedOrDeleted1SizedDimensions() local1037 for (size_t i = 0; i <= unmodified_dims.size(); ++i) { in InsertedOrDeleted1SizedDimensions()1041 i > 0 ? unmodified_dims[i - 1] : std::make_pair(-1LL, -1LL); in InsertedOrDeleted1SizedDimensions()1043 i < unmodified_dims.size() in InsertedOrDeleted1SizedDimensions()1044 ? unmodified_dims[i] in InsertedOrDeleted1SizedDimensions()