Searched refs:infeed_slices (Results 1 – 2 of 2) sorted by relevance
26 const ShapeTree<BufferAllocation::Slice>& infeed_slices, in InfeedThunk() argument28 : Thunk(Kind::kInfeed, hlo_instruction), infeed_slices_(infeed_slices) {} in InfeedThunk()
37 InfeedThunk(const ShapeTree<BufferAllocation::Slice>& infeed_slices,