| /external/tensorflow/tensorflow/stream_executor/ |
| D | scratch_allocator.h | 54 virtual port::StatusOr<DeviceMemory<uint8>> AllocateBytes(
70 port::StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream,
|
| D | scratch_allocator.cc | 32 port::StatusOr<DeviceMemory<uint8>> OneTimeScratchAllocator::AllocateBytes( in AllocateBytes() function in stream_executor::OneTimeScratchAllocator
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | scratch_allocator.h | 40 StatusOr<se::DeviceMemory<uint8>> AllocateBytes(se::Stream* stream,
47 AllocateBytes(stream, num_elements * sizeof(T))); in Allocate()
|
| D | scratch_allocator.cc | 21 StatusOr<se::DeviceMemory<uint8>> ScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::ScratchAllocator
|
| D | fft_thunk.h | 47 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes(
|
| D | cudnn_conv_algorithm_picker.cc | 214 input_output_allocator.AllocateBytes( in PickBestAlgorithm()
221 input_output_allocator.AllocateBytes( in PickBestAlgorithm()
|
| D | fft_thunk.cc | 40 StatusOr<se::DeviceMemory<uint8>> FftScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::FftScratchAllocator
|
| D | cudnn_conv_runner.cc | 91 se::port::StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in xla::gpu::__anon2b86fa710111::ScratchBufAllocator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 47 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function
|
| D | cudnn_rnn_ops.cc | 370 StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream, in AllocateBytes() function in tensorflow::__anon454a70830111::CudnnRnnAllocatorInTemp
415 StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream, in AllocateBytes() function in tensorflow::__anon454a70830111::CudnnRnnAllocatorInOutput
456 StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream, in AllocateBytes() function in tensorflow::__anon454a70830111::CudnnRNNPersistentSpaceAllocator
|
| D | fft_ops.cc | 319 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anona0c96f1e0111::CufftScratchAllocator
|
| D | batch_matmul_op_impl.h | 265 se::port::StatusOr<DeviceMemoryBytes> AllocateBytes(
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1985 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in CreateRnnWorkspace()
2062 auto allocated = reserve_space_allocator->AllocateBytes( in DoRnnForwardImpl()
2575 mac->scratch_allocator_->AllocateBytes(mac->stream_, size_in_bytes); in MIOpenAllocatorCallback()
2658 auto allocated = scratch_allocator->AllocateBytes(stream, size_in_bytes); in DoPrepareForConvolution()
2749 scratch_allocator->AllocateBytes(stream, scratch_memory_size); in DoPrepareForConvolution()
3593 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in DoPoolBackward()
3617 auto allocated = workspace_allocator->AllocateBytes(stream, dest2_size); in DoPoolBackward()
3689 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in DoPoolBackward()
3713 auto allocated = workspace_allocator->AllocateBytes(stream, dest2_size); in DoPoolBackward()
3824 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in DoNormalizeBackwardWithDimensions()
[all …]
|
| D | rocm_fft.cc | 264 scratch_allocator->AllocateBytes(stream, size_in_bytes); in Initialize()
320 scratch_allocator->AllocateBytes(stream, size_in_bytes); in Initialize()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 948 SE_ASSIGN_OR_RETURN(state_memory, state_allocator->AllocateBytes( in Create()
1500 return workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in CreateRnnWorkspace()
1549 reserve_space_allocator->AllocateBytes( in DoRnnForwardImpl()
2248 return scratch_allocator->AllocateBytes(stream, size_in_bytes); in AllocateCudnnConvolutionForwardWorkspace()
2293 return scratch_allocator->AllocateBytes(stream, size_in_bytes); in AllocateCudnnConvolutionBackwardDataWorkspace()
2338 return scratch_allocator->AllocateBytes(stream, size_in_bytes); in AllocateCudnnConvolutionBackwardFilterWorkspace()
|
| D | cuda_fft.cc | 248 scratch_allocator->AllocateBytes(stream, scratch_size_bytes_); in UpdateScratchAllocator()
|
| D | cuda_blas.cc | 2182 scratch_allocator->AllocateBytes(stream, size)); in DoBlasGemmBatchedInternal()
2184 scratch_allocator->AllocateBytes(stream, size)); in DoBlasGemmBatchedInternal()
2186 scratch_allocator->AllocateBytes(stream, size)); in DoBlasGemmBatchedInternal()
|
| /external/protobuf/src/google/protobuf/ |
| D | descriptor.cc | 552 void* AllocateBytes(int size);
983 return reinterpret_cast<Type*>(AllocateBytes(sizeof(Type))); in Allocate()
988 return reinterpret_cast<Type*>(AllocateBytes(sizeof(Type) * count)); in AllocateArray()
1010 void* DescriptorPool::Tables::AllocateBytes(int size) { in AllocateBytes() function in google::protobuf::DescriptorPool::Tables
|