Searched refs:body_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 159 TfLiteTensor* body_input = in Prepare() local
163 TF_LITE_ENSURE_EQ(context, body_input->type, body_output->type); in Prepare()
167 if (!TfLiteIntArrayEqual(body_input->dims, body_output->dims)) { in Prepare()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 212 auto body_input = GetTupleElement(input_tuple, 2); in InvertDiagonalBlocks() local
217 DynamicSlice(body_input, {zero, j, zero}, in InvertDiagonalBlocks()
234 Tuple(bodyb.get(), {next_i, body_out, body_input}); in InvertDiagonalBlocks()
|