Searched defs:body_wrapper (Results 1 – 2 of 2) sorted by relevance
101 def body_wrapper(*inputs): function217 def body_wrapper(i, *args): function
220 xla::XlaComputation body_wrapper; in BuildWrappedBody() local