Home
last modified time | relevance | path

Searched refs:HloBufferSlice (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h48 struct HloBufferSlice : public BufferSlice { struct
Dir_emitter_unnested.cc3240 static std::vector<HloBufferSlice> GetHloBufferSlices( in GetHloBufferSlices()
3242 std::vector<HloBufferSlice> result; in GetHloBufferSlices()
3317 HloBufferSlice hlo_buffer_slice; in GetHloBufferSlices()
3457 std::vector<HloBufferSlice> hlo_slices = in BuildKernelThunk()
3470 const HloBufferSlice* hlo_buffer_slice = in BuildKernelThunk()
3471 static_cast<const HloBufferSlice*>(slice); in BuildKernelThunk()