Searched defs:unmodified_dims (Results 1 – 4 of 4) sorted by relevance
488 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in MatchReduceScatter() local
2447 auto unmodified_dims = ShapeUtil::DimensionsUnmodifiedByReshape( in OptimizeDotOfReorderContractingDims() local5447 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in HandleReduce() local
1230 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in InsertedOrDeleted1SizedDimensions() local1285 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in ReshapeLeavesDimensionsUnmodified() local
661 std::vector<std::pair<int64_t, int64_t>> unmodified_dims = in AddBroadcastSequence() local