Searched refs:BuildInfeedThunk (Results 1 – 2 of 2) sorted by relevance
347 std::unique_ptr<Thunk> BuildInfeedThunk(const HloInstruction* inst);
1474 AddThunkToThunkSequence(BuildInfeedThunk(infeed)); in HandleInfeed()1728 std::unique_ptr<Thunk> IrEmitterUnnested::BuildInfeedThunk( in BuildInfeedThunk() function in xla::gpu::IrEmitterUnnested