Searched refs:updated_values (Results 1 – 4 of 4) sorted by relevance
97 std::vector<XlaOp> updated_values; in ForEachIndex() local98 updated_values.reserve(values.size()); in ForEachIndex()99 updated_values.push_back(Add( in ForEachIndex()106 updated_values.insert(updated_values.end(), body_outputs.begin(), in ForEachIndex()108 return updated_values; in ForEachIndex()
299 std::vector<XlaOp> updated_values; in HouseHolderBidiagonalization() local300 updated_values.reserve(values.size()); in HouseHolderBidiagonalization()302 updated_values.push_back(i + one); in HouseHolderBidiagonalization()303 updated_values.push_back(u); in HouseHolderBidiagonalization()304 updated_values.push_back(v); in HouseHolderBidiagonalization()305 updated_values.push_back(a); in HouseHolderBidiagonalization()306 updated_values.push_back(eps); in HouseHolderBidiagonalization()307 return updated_values; in HouseHolderBidiagonalization()698 std::vector<XlaOp> updated_values; in WhileLoopFn() local699 updated_values.reserve(values_inner.size()); in WhileLoopFn()[all …]
323 std::vector<XlaOp> updated_values; in WhileLoopFn() local324 updated_values.reserve(values_inner.size()); in WhileLoopFn()326 updated_values.push_back(k + ScalarLike(k, 1)); in WhileLoopFn()327 updated_values.push_back(values_inner[1]); in WhileLoopFn()328 updated_values.push_back(values_inner[2]); in WhileLoopFn()329 updated_values.push_back(values_inner[3]); in WhileLoopFn()331 return updated_values; in WhileLoopFn()
1693 std::vector<XlaOp> updated_values(values.size()); in LentzThompsonBarnettAlgorithm() local1694 updated_values[kIterationIdx] = Add(iteration, ScalarLike(iteration, 1)); in LentzThompsonBarnettAlgorithm()1695 updated_values[kCIdx] = c; in LentzThompsonBarnettAlgorithm()1696 updated_values[kDIdx] = d; in LentzThompsonBarnettAlgorithm()1697 updated_values[kHIdx] = h; in LentzThompsonBarnettAlgorithm()1699 updated_values.begin() + kFirstInputIdx); in LentzThompsonBarnettAlgorithm()1704 updated_values[kValuesUnconvergedIdx] = in LentzThompsonBarnettAlgorithm()1707 return updated_values; in LentzThompsonBarnettAlgorithm()