Searched refs:hlo_buffer_slice (Results 1 – 1 of 1) sorted by relevance
3317 HloBufferSlice hlo_buffer_slice; in GetHloBufferSlices() local3318 hlo_buffer_slice.instr = instr; in GetHloBufferSlices()3319 hlo_buffer_slice.hlo_index = index; in GetHloBufferSlices()3320 hlo_buffer_slice.buffer_slice = maybe_slice->first; in GetHloBufferSlices()3321 hlo_buffer_slice.gte_index = maybe_slice->second; in GetHloBufferSlices()3322 result.push_back(hlo_buffer_slice); in GetHloBufferSlices()3470 const HloBufferSlice* hlo_buffer_slice = in BuildKernelThunk() local3472 const HloInstruction* instr = hlo_buffer_slice->instr; in BuildKernelThunk()3473 const ShapeIndex& index = hlo_buffer_slice->hlo_index; in BuildKernelThunk()3476 << hlo_buffer_slice->buffer_slice.ToString() << " at GTE index " in BuildKernelThunk()[all …]