Home
last modified time | relevance | path

Searched defs:buffer_allocations (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinfeed_thunk.cc35 const BufferAllocations& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dconvolution_thunk.cc56 const auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Doutfeed_thunk.cc34 const BufferAllocations& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dthunk.cc23 const BufferAllocations& buffer_allocations, se::Stream* stream, in ExecuteParams()
Dtriangular_solve_thunk.cc76 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dthunk.h111 const BufferAllocations* buffer_allocations; // never null member
Dfft_thunk.cc86 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dgpu_executable.cc321 const BufferAllocations& buffer_allocations, in ExecuteThunks()
575 const BufferAllocations& buffer_allocations, in ExecuteJitRt()
846 const BufferAllocations& buffer_allocations, bool block_host_until_done) { in ExecuteThunksOrJitRt()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo.proto624 repeated BufferAllocationProto buffer_allocations = 3; field