Searched refs:unmodified_dims (Results 1 – 4 of 4) sorted by relevance
1025 std::vector<std::pair<int64, int64>> unmodified_dims = in InsertedOrDeleted1SizedDimensions() local1031 for (size_t i = 0; i <= unmodified_dims.size(); ++i) { in InsertedOrDeleted1SizedDimensions()1035 i > 0 ? unmodified_dims[i - 1] : std::make_pair(-1LL, -1LL); in InsertedOrDeleted1SizedDimensions()1037 i < unmodified_dims.size() in InsertedOrDeleted1SizedDimensions()1038 ? unmodified_dims[i] in InsertedOrDeleted1SizedDimensions()
335 std::vector<std::pair<int64, int64>> unmodified_dims = in HandleReshape() local338 for (auto& unmodified : unmodified_dims) { in HandleReshape()
1973 std::vector<std::pair<int64, int64>> unmodified_dims = in ReshapeLeavesDimensionsUnmodified() local1980 while (i < unmodified_dims.size() && in ReshapeLeavesDimensionsUnmodified()1981 unmodified_dims[i].first < input_dim_index) { in ReshapeLeavesDimensionsUnmodified()1984 if (i >= unmodified_dims.size() || in ReshapeLeavesDimensionsUnmodified()1985 unmodified_dims[i].first != input_dim_index) { in ReshapeLeavesDimensionsUnmodified()1988 output_dim_indices.push_back(unmodified_dims[i].second); in ReshapeLeavesDimensionsUnmodified()3019 std::vector<std::pair<int64, int64>> unmodified_dims = in HandleReduce() local3027 for (auto dim_pair : unmodified_dims) { in HandleReduce()3042 for (auto dim_pair : unmodified_dims) { in HandleReduce()
2707 std::vector<std::pair<int64, int64>> unmodified_dims = in InferReshapeShape() local2709 for (auto& unmodified : unmodified_dims) { in InferReshapeShape()