Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc51 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
88 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
134 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dconvolution_thunk.cc43 const auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dinfeed_thunk.cc35 const BufferAllocations& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Doutfeed_thunk.cc34 const BufferAllocations& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dtriangular_solve_thunk.cc72 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dfft_thunk.cc118 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
Dthunk.h108 const BufferAllocations* buffer_allocations; // never null member
Dgpu_executable.cc151 const BufferAllocations& buffer_allocations, bool block_host_until_done) { in ExecuteThunks()
522 const BufferAllocations& buffer_allocations, in ExecuteBef()
Dnccl_all_reduce_thunk.cc41 const BufferAllocations& buffer_allocations, in RunAllReduce()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo.proto551 repeated BufferAllocationProto buffer_allocations = 3; field