| /external/tensorflow/tensorflow/stream_executor/ |
| D | scratch_allocator.h | 55 virtual port::StatusOr<DeviceMemory<uint8>> AllocateBytes(
71 port::StatusOr<DeviceMemory<uint8>> AllocateBytes(Stream* stream,
|
| D | scratch_allocator.cc | 33 port::StatusOr<DeviceMemory<uint8>> OneTimeScratchAllocator::AllocateBytes( in AllocateBytes() function in perftools::gputools::OneTimeScratchAllocator
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_convolution_algorithm_picker.cc | 47 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes(
68 se::port::StatusOr<se::DeviceMemory<uint8>> ScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::__anon5d1dd1190111::ScratchAllocator
197 input_output_allocator.AllocateBytes(&stream, in PickBestAlgorithm()
200 input_output_allocator.AllocateBytes(&stream, in PickBestAlgorithm()
203 input_output_allocator.AllocateBytes(&stream, in PickBestAlgorithm()
|
| D | fft_thunk.h | 49 AllocateBytes(perftools::gputools::Stream* stream, int64 byte_size) override;
|
| D | fft_thunk.cc | 52 se::port::StatusOr<se::DeviceMemory<uint8>> FftScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::FftScratchAllocator
|
| D | cudnn_convolution_runner.cc | 52 se::port::StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in xla::gpu::__anon6ce1464f0111::ScratchBufAllocator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 48 AllocateBytes(perftools::gputools::Stream* stream, int64 byte_size) override { in AllocateBytes() function
|
| D | fft_ops.cc | 310 gpu::port::StatusOr<gpu::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anon2cde7a920111::CufftScratchAllocator
|
| D | batch_matmul_op_impl.h | 263 perftools::gputools::port::StatusOr<DeviceMemoryBytes> AllocateBytes(
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| D | cudnn_rnn_ops.cc | 240 StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anonc95c99b30111::CudnnRNNWorkspaceAllocator
276 StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anonc95c99b30111::CudnnRNNReserveSpaceAllocator
317 StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anonc95c99b30111::CudnnRNNPersistentSpaceAllocator
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 998 state_allocator->AllocateBytes(nullptr, state_sizes_in_bytes); in CudnnDropoutDescriptor()
1518 workspace_allocator->AllocateBytes(stream, workspace_size_in_bytes); in CreateRnnWorkspace()
1594 auto allocated = reserve_space_allocator->AllocateBytes( in DoRnnForwardImpl()
2188 allocated = scratch_allocator->AllocateBytes(stream, size_in_bytes); in GetCudnnConvolutionForwardAlgorithm()
2381 scratch_allocator->AllocateBytes(stream, size_in_bytes); in DoConvolveImpl()
2427 auto allocated = scratch_allocator->AllocateBytes(stream, size_in_bytes); in DoConvolveImpl()
3225 scratch_allocator->AllocateBytes(stream, size_in_bytes); in DoConvolveBackwardDataImpl()
3273 auto allocated = scratch_allocator->AllocateBytes(stream, size_in_bytes); in DoConvolveBackwardDataImpl()
3474 scratch_allocator->AllocateBytes(stream, size_in_bytes); in DoConvolveBackwardFilterImpl()
3520 auto allocated = scratch_allocator->AllocateBytes(stream, size_in_bytes); in DoConvolveBackwardFilterImpl()
|
| D | cuda_fft.cc | 287 scratch_allocator->AllocateBytes(stream, scratch_size_bytes_); in UpdateScratchAllocator()
|
| D | cuda_blas.cc | 2351 scratch_allocator->AllocateBytes(stream, size)); in DoBlasGemmBatchedInternal()
2353 scratch_allocator->AllocateBytes(stream, size)); in DoBlasGemmBatchedInternal()
2355 scratch_allocator->AllocateBytes(stream, size)); in DoBlasGemmBatchedInternal()
|
| /external/v8/src/heap/ |
| D | spaces.h | 1607 void AllocateBytes(size_t bytes) { in AllocateBytes() function
2049 accounting_stats_.AllocateBytes(bytes); in NON_EXPORTED_BASE()
|
| /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
|