Searched defs:AllocateBytes (Results 1 – 8 of 8) sorted by relevance
21 StatusOr<se::DeviceMemory<uint8>> ScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::ScratchAllocator
40 StatusOr<se::DeviceMemory<uint8>> FftScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::FftScratchAllocator
91 se::port::StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in xla::gpu::__anon2b86fa710111::ScratchBufAllocator
32 port::StatusOr<DeviceMemory<uint8>> OneTimeScratchAllocator::AllocateBytes( in AllocateBytes() function in stream_executor::OneTimeScratchAllocator
47 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function
319 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anona0c96f1e0111::CufftScratchAllocator
370 StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream, in AllocateBytes() function in tensorflow::__anon454a70830111::CudnnRnnAllocatorInTemp415 StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream, in AllocateBytes() function in tensorflow::__anon454a70830111::CudnnRnnAllocatorInOutput456 StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream, in AllocateBytes() function in tensorflow::__anon454a70830111::CudnnRNNPersistentSpaceAllocator
1010 void* DescriptorPool::Tables::AllocateBytes(int size) { in AllocateBytes() function in google::protobuf::DescriptorPool::Tables