Searched refs:BuildWhileThunk (Results 1 – 2 of 2) sorted by relevance
687 StatusOr<std::unique_ptr<Thunk>> BuildWhileThunk(const HloInstruction* hlo);
2444 TF_ASSIGN_OR_RETURN(auto thunk, BuildWhileThunk(xla_while)); in HandleWhile()3834 StatusOr<std::unique_ptr<Thunk>> IrEmitterUnnested::BuildWhileThunk( in BuildWhileThunk() function in xla::gpu::IrEmitterUnnested