Searched refs:BufferAllocations (Results 1 – 25 of 39) sorted by relevance
12
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 35 class BufferAllocations {
51 StatusOr<std::unique_ptr<BufferAllocations>> Build(
60 ~BufferAllocations();
62 BufferAllocations(const BufferAllocations&) = delete;
63 BufferAllocations& operator=(const BufferAllocations&) = delete;
86 BufferAllocations(BufferAllocation::Index buffer_count, int device_ordinal, in BufferAllocations() function
|
| D | buffer_allocations.cc | 35 void BufferAllocations::Builder::RegisterBuffer(BufferAllocation::Index index, in RegisterBuffer()
40 StatusOr<std::unique_ptr<BufferAllocations>> BufferAllocations::Builder::Build( in Build()
44 auto buffer_allocations = absl::WrapUnique(new BufferAllocations( in Build()
118 BufferAllocations::~BufferAllocations() { in ~BufferAllocations()
128 Status BufferAllocations::TearDown( in TearDown()
153 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress()
160 se::DeviceMemoryBase BufferAllocations::GetDeviceAddress( in GetDeviceAddress()
170 void BufferAllocations::SetBuffer(BufferAllocation::Index buffer_index, in SetBuffer()
|
| D | memset_thunk.cc | 25 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
34 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | cudnn_batchnorm_thunk.h | 63 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
94 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
128 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | memset_thunk.h | 39 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
56 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | copy_thunk.h | 43 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
67 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | copy_thunk.cc | 34 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
53 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | cudnn_batchnorm_thunk.cc | 102 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
164 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
248 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | outfeed_thunk.h | 41 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | infeed_thunk.h | 43 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | nccl_all_reduce_thunk.h | 48 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | for_thunk.h | 42 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | sequential_thunk.h | 44 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | tuple_thunk.h | 48 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | while_thunk.h | 51 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | sequential_thunk.cc | 37 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | convolution_thunk.h | 56 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | nccl_all_reduce_thunk.cc | 315 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
335 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | conditional_thunk.h | 56 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | tuple_thunk.cc | 25 Status TupleThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | for_thunk.cc | 43 Status ForThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | cholesky_thunk.h | 54 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | triangular_solve_thunk.h | 51 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | gemm_thunk.h | 52 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
| D | kernel_thunk.h | 65 Status ExecuteOnStream(const BufferAllocations& buffer_allocations,
|
12