Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h33 struct BufferSlice { struct
48 struct HloBufferSlice : public BufferSlice { argument
53 struct MlirBufferSlice : public BufferSlice {
643 absl::Span<const BufferSlice* const> slices,
644 std::function<void(const BufferSlice*, llvm::Value*)>
Dir_emitter_unnested.cc3343 absl::Span<const BufferSlice* const> slices, in BuildKernelThunkFromBufferSlices()
3344 std::function<void(const BufferSlice*, llvm::Value*)> in BuildKernelThunkFromBufferSlices() argument
3460 std::vector<BufferSlice*> slice_ptrs; in BuildKernelThunk()
3469 slice_ptrs, [this](const BufferSlice* slice, llvm::Value* value) { in BuildKernelThunk()
3488 std::vector<const BufferSlice*> slice_ptrs; in BuildKernelThunkForMlirImpl()
3500 [&](const BufferSlice* slice, llvm::Value* value) { in BuildKernelThunkForMlirImpl()