Searched refs:unmodified_dims (Results 1 – 3 of 3) 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::pair<int64, int64>(-1, -1); in InsertedOrDeleted1SizedDimensions()1037 i < unmodified_dims.size() in InsertedOrDeleted1SizedDimensions()1038 ? unmodified_dims[i] in InsertedOrDeleted1SizedDimensions()1078 std::vector<std::pair<int64, int64>> unmodified_dims = in ReshapeLeavesDimensionsUnmodified() local1084 while (i < unmodified_dims.size() && in ReshapeLeavesDimensionsUnmodified()1085 unmodified_dims[i].first < input_dim_index) { in ReshapeLeavesDimensionsUnmodified()1088 if (i >= unmodified_dims.size() || in ReshapeLeavesDimensionsUnmodified()1089 unmodified_dims[i].first != input_dim_index) { in ReshapeLeavesDimensionsUnmodified()[all …]
1684 auto unmodified_dims = ShapeUtil::DimensionsUnmodifiedByReshape( in OptimizeDotOfReorderContractingDims() local1687 if ((unmodified_dims.size() != reshape->shape().rank() - 1) || in OptimizeDotOfReorderContractingDims()1688 absl::c_any_of(unmodified_dims, [&](const std::pair<int64, int64>& p) { in OptimizeDotOfReorderContractingDims()1702 for (const auto& pair : unmodified_dims) { in OptimizeDotOfReorderContractingDims()3661 std::vector<std::pair<int64, int64>> unmodified_dims = in HandleReduce() local3669 for (auto dim_pair : unmodified_dims) { in HandleReduce()3684 for (auto dim_pair : unmodified_dims) { in HandleReduce()
520 std::vector<std::pair<int64, int64>> unmodified_dims = in AddBroadcastSequence() local523 for (auto& unmodified : unmodified_dims) { in AddBroadcastSequence()