Searched defs:AllocateBytes (Results 1 – 12 of 12) sorted by relevance
31 port::StatusOr<DeviceMemory<uint8>> OneTimeScratchAllocator::AllocateBytes( in AllocateBytes() function in stream_executor::OneTimeScratchAllocator
45 void* Allocator::AllocateBytes(std::ptrdiff_t num_bytes) { in AllocateBytes() function in ruy::Allocator
57 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function
368 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anon049b3f3c0111::CufftScratchAllocator
675 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp728 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
370 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInTemp414 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInOutput454 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::__anond88b5f210111::CudnnRNNPersistentSpaceAllocator
40 StatusOr<se::DeviceMemory<uint8>> FftScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::FftScratchAllocator
54 se::port::StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in xla::gpu::__anon53f8f4230111::ScratchBufAllocator
81 StatusOr<se::DeviceMemory<uint8>> ScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::__anon6437d7ed0111::ScratchAllocator
63 char* AllocateBytes(size_t size) { in AllocateBytes() function
63 port::StatusOr<DeviceMemory<uint8>> RedzoneAllocator::AllocateBytes( in AllocateBytes() function in stream_executor::RedzoneAllocator
1226 void* DescriptorPool::Tables::AllocateBytes(int size) { in AllocateBytes() function in google::protobuf::DescriptorPool::Tables