Searched defs:buffer_allocations (Results 1 – 10 of 10) sorted by relevance
51 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local88 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local134 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
43 const auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
35 const BufferAllocations& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
34 const BufferAllocations& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
72 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
118 auto& buffer_allocations = *params.buffer_allocations; in ExecuteOnStream() local
108 const BufferAllocations* buffer_allocations; // never null member
151 const BufferAllocations& buffer_allocations, bool block_host_until_done) { in ExecuteThunks()522 const BufferAllocations& buffer_allocations, in ExecuteBef()
41 const BufferAllocations& buffer_allocations, in RunAllReduce()
551 repeated BufferAllocationProto buffer_allocations = 3; field