Searched refs:updated_values (Results 1 – 3 of 3) 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()
307 std::vector<XlaOp> updated_values; in HouseHolderBidiagonalization() local308 updated_values.reserve(values.size()); in HouseHolderBidiagonalization()310 updated_values.push_back(i + one); in HouseHolderBidiagonalization()311 updated_values.push_back(u); in HouseHolderBidiagonalization()312 updated_values.push_back(v); in HouseHolderBidiagonalization()313 updated_values.push_back(a); in HouseHolderBidiagonalization()314 updated_values.push_back(eps); in HouseHolderBidiagonalization()315 return updated_values; in HouseHolderBidiagonalization()708 std::vector<XlaOp> updated_values; in WhileLoopFn() local709 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()