Searched defs:buffer_allocations (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | copy_thunk.cc | 33 const BufferAllocations& buffer_allocations, in ExecuteOnStream()
51 const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | tuple_thunk.cc | 26 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
|
| D | cudnn_batchnorm_thunk.cc | 103 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
162 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
244 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
|
| D | for_thunk.cc | 39 const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | sequential_thunk.cc | 36 const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | while_thunk.cc | 43 Status WhileThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | conditional_thunk.cc | 45 const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | infeed_thunk.cc | 33 Status InfeedThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | kernel_thunk.cc | 66 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
|
| D | convolution_thunk.cc | 65 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
|
| D | buffer_allocations.cc | 45 auto buffer_allocations = WrapUnique( in Build() local
|
| D | fft_thunk.cc | 127 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
|
| D | gpu_executable.cc | 131 const BufferAllocations& buffer_allocations, bool block_host_until_done, in ExecuteThunks()
|
| D | gemm_thunk.cc | 227 const BufferAllocations& buffer_allocations, se::Stream* stream) { in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 259 repeated BufferAllocationProto buffer_allocations = 3; field
|