Searched defs:wrapped_body (Results 1 – 2 of 2) sorted by relevance
243 const xla::XlaComputation& wrapped_body, in BuildWhile()570 xla::XlaComputation wrapped_body = std::move(body_result.ValueOrDie()); in Compile() local
155 def wrapped_body(loop_counter, maximum_iterations_arg, *args): function