Searched refs:while_result (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | while_op.cc | 231 xla::XlaOp while_result = in BuildWhile() local
238 xla::GetTupleElement(while_result, while_result_index); in BuildWhile()
247 while_result, ctx->num_inputs() - num_compile_time_const_args)); in BuildWhile()
524 xla::XlaOp while_result = in Compile() local
533 ctx->SetOutput(i, xla::GetTupleElement(while_result, i)); in Compile()
542 xla::GetTupleElement(while_result, ctx->num_outputs()); in Compile()
562 xla::GetTupleElement(while_result, pos), builder)); in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 303 auto while_result = While(condition, body, initial_while_state); in XLA_TEST_F() local
304 Add(GetTupleElement(while_result, 1), in XLA_TEST_F()
|