Searched defs:buffer_allocations (Results 1 – 20 of 20) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| 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 | 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 | 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 | sequential_thunk.cc | 37 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | while_thunk.cc | 50 Status WhileThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | convolution_thunk.cc | 45 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | conditional_thunk.cc | 60 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | outfeed_thunk.cc | 32 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | cholesky_thunk.cc | 54 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| 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 | kernel_thunk.cc | 74 Status KernelThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | triangular_solve_thunk.cc | 72 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteOnStream()
|
| D | infeed_thunk.cc | 30 Status InfeedThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | buffer_allocations.cc | 44 auto buffer_allocations = absl::WrapUnique(new BufferAllocations( in Build() local
|
| D | fft_thunk.cc | 109 Status FftThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| D | gpu_executable.cc | 68 const BufferAllocations& buffer_allocations, bool block_host_until_done, in ExecuteThunks()
|
| D | gemm_thunk.cc | 311 Status GemmThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo.proto | 444 repeated BufferAllocationProto buffer_allocations = 3; field
|