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/
Dcudnn_batchnorm_thunk.cc52 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
91 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
139 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dconvolution_thunk.cc44 const auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dinfeed_thunk.cc35 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Doutfeed_thunk.cc35 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dtriangular_solve_thunk.cc73 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dfft_thunk.cc119 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dthunk.h106 const BufferAllocations* buffer_allocations; // never null member
Dgpu_executable.cc130 const BufferAllocations& buffer_allocations, bool block_host_until_done, in ExecuteThunks()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo.proto502 repeated BufferAllocationProto buffer_allocations = 3; field