Searched refs:body_out (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 211 auto body_out = GetTupleElement(input_tuple, 1); in InvertDiagonalBlocks() local
229 auto update = -DotGeneral(input_row, body_out, dnums, &precision_proto); in InvertDiagonalBlocks()
231 body_out = DynamicUpdateSlice(body_out, update, {zero, j, zero}); in InvertDiagonalBlocks()
234 Tuple(bodyb.get(), {next_i, body_out, body_input}); in InvertDiagonalBlocks()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2.py | 262 grads = [_preprocess_grad(grad, body_out, while_out)
263 for grad, body_out, while_out
|