Searched refs:body_l (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 92 auto body_l = loop_vars[1]; in CholeskyUnblocked() local
97 auto l_i = DynamicSliceInMinorDims(body_l, {i, zero}, {1, n}); in CholeskyUnblocked()
117 auto dot = BatchDot(body_l, TransposeInMinorDims(row), precision); in CholeskyUnblocked()
121 body_l = in CholeskyUnblocked()
122 DynamicUpdateSliceInMinorDims(body_l, (a_ip1i - dot_ip1) / l_ii, {i}); in CholeskyUnblocked()
125 body_l = DynamicUpdateSliceInMinorDims(body_l, l_ii, {i, i}); in CholeskyUnblocked()
127 return std::vector<XlaOp>{body_a, body_l}; in CholeskyUnblocked()
|