Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtuple_thunk.cc27 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dcudnn_batchnorm_thunk.cc119 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
169 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
237 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dconvolution_thunk.cc45 const auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Doutfeed_thunk.cc33 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dtriangular_solve_thunk.cc73 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dinfeed_thunk.cc32 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dgpu_executable.cc140 const BufferAllocations& buffer_allocations, bool block_host_until_done, in ExecuteThunks()
327 std::unique_ptr<BufferAllocations> buffer_allocations; in ExecuteAsyncOnStream() local
Dbuffer_allocations.cc44 auto buffer_allocations = absl::WrapUnique(new BufferAllocations( in Build() local
Dcustom_call_thunk.cc58 const BufferAllocations* buffer_allocations, in AppendBuffersFor()
Dfft_thunk.cc111 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dthunk.h95 const BufferAllocations* buffer_allocations; // never null member
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo.proto476 repeated BufferAllocationProto buffer_allocations = 3; field