| /external/ruy/ruy/ |
| D | allocator.cc | 45 void* Allocator::AllocateBytes(std::ptrdiff_t num_bytes) { in AllocateBytes() function in ruy::Allocator
78 void* p = AllocateBytes(num_bytes + kMinPeriod); in AllocateBytesAvoidingAliasingWith()
96 return AllocateBytes(num_bytes); in AllocateBytesAvoidingAliasingWith()
|
| D | allocator.h | 64 void* AllocateBytes(std::ptrdiff_t num_bytes);
72 return static_cast<T*>(AllocateBytes(count * sizeof(T))); in Allocate()
|
| D | allocator_test.cc | 76 allocator.AllocateBytes(1); in TEST()
94 allocator.AllocateBytes(1); in TEST()
|
| D | prepare_packed_matrices.cc | 89 packed_matrix.sums = allocator->AllocateBytes(SumsBytes(packed_matrix)); in PreparePackedMatrices()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | scratch_allocator.h | 50 virtual port::StatusOr<DeviceMemory<uint8>> AllocateBytes(
66 port::StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override;
|
| D | scratch_allocator.cc | 31 port::StatusOr<DeviceMemory<uint8>> OneTimeScratchAllocator::AllocateBytes( in AllocateBytes() function in stream_executor::OneTimeScratchAllocator
|
| /external/libtextclassifier/native/utils/base/ |
| D | arena.cc | 63 char* AllocateBytes(size_t size) { in AllocateBytes() function
94 char* AllocateBytes(size_t size) {
153 first_blocks_[0].mem = AllocateBytes(block_size_); in BaseArena()
322 block->mem = AllocateBytes(adjusted_block_size); in AllocNewBlock()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 65 StatusOr<se::DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override;
70 AllocateBytes(num_elements * sizeof(T))); in Allocate()
81 StatusOr<se::DeviceMemory<uint8>> ScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::__anon6437d7ed0111::ScratchAllocator
408 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheCuda()
414 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheCuda()
568 input_output_allocator.AllocateBytes(result_buffer.size())); in PickBestAlgorithmNoCacheCuda()
677 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheRocm()
685 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheRocm()
|
| D | fft_thunk.h | 48 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes(
|
| D | gemm_algorithm_picker.cc | 80 input_output_allocator.AllocateBytes( in DoUncachedGemmAutotune()
|
| D | fft_thunk.cc | 40 StatusOr<se::DeviceMemory<uint8>> FftScratchAllocator::AllocateBytes( in AllocateBytes() function in xla::gpu::FftScratchAllocator
|
| D | gpu_conv_runner.cc | 54 se::port::StatusOr<DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in xla::gpu::__anon53f8f4230111::ScratchBufAllocator
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 67 allocator.AllocateBytes(/*byte_size=*/kAllocSize)); in TEST()
140 (void)allocator.AllocateBytes(/*byte_size=*/1); in TEST()
|
| D | cuda_fft.cc | 251 auto allocated = scratch_allocator->AllocateBytes(scratch_size_bytes_); in UpdateScratchAllocator()
483 auto allocated = allocator->AllocateBytes(input.size()); in DoFftInternal()
|
| D | cuda_dnn.cc | 1008 state_allocator->AllocateBytes(state_sizes_in_bytes)); in Create()
1689 return workspace_allocator->AllocateBytes(workspace_size_in_bytes); in CreateRnnWorkspace()
1714 return workspace_allocator->AllocateBytes(workspace_size_in_bytes); in CreateBatchNormForwardWorkspace()
1738 return workspace_allocator->AllocateBytes(workspace_size_in_bytes); in CreateBatchNormBackwardWorkspace()
1788 SE_ASSIGN_OR_RETURN(reserve_space, reserve_space_allocator->AllocateBytes( in DoRnnForwardImpl()
2574 return scratch_allocator->AllocateBytes(size_in_bytes); in AllocateCudnnConvolutionForwardWorkspace()
2620 return scratch_allocator->AllocateBytes(size_in_bytes); in AllocateCudnnConvolutionBackwardDataWorkspace()
2666 return scratch_allocator->AllocateBytes(size_in_bytes); in AllocateCudnnConvolutionBackwardFilterWorkspace()
3552 SE_ASSIGN_OR_RETURN(reserve_space, reserve_space_allocator->AllocateBytes( in DoBatchNormalizationForwardImpl()
3946 scratch_allocator->AllocateBytes(workspace_size_in_bytes); in DoPrepareForCtcLoss()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.h | 59 port::StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override;
|
| D | redzone_allocator.cc | 63 port::StatusOr<DeviceMemory<uint8>> RedzoneAllocator::AllocateBytes( in AllocateBytes() function in stream_executor::RedzoneAllocator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_gpu.h | 57 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function
|
| D | gpu_utils.cc | 45 auto output_rz_or = rz_allocator->AllocateBytes(buffer.size()); in WrapRedzoneBestEffort()
|
| D | cudnn_rnn_ops.cc | 370 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInTemp
414 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInOutput
454 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::__anond88b5f210111::CudnnRNNPersistentSpaceAllocator
|
| D | fft_ops.cc | 368 se::port::StatusOr<se::DeviceMemory<uint8>> AllocateBytes( in AllocateBytes() function in tensorflow::__anon049b3f3c0111::CufftScratchAllocator
|
| D | fused_batch_norm_op.cc | 675 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp
728 StatusOr<DeviceMemory<uint8>> AllocateBytes(int64 byte_size) override { in AllocateBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
|
| D | matmul_op_impl.h | 297 se::port::StatusOr<DeviceMemoryBytes> AllocateBytes(
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2066 workspace_allocator->AllocateBytes(workspace_size_in_bytes); in CreateRnnWorkspace()
2144 reserve_space_allocator->AllocateBytes(reserve_space_size_in_bytes); in DoRnnForwardImpl()
2434 auto scratch_or = scratch_allocator->AllocateBytes(workspace_size_in_bytes); in DoPrepareForCtcLoss()
2827 auto allocated = mac->scratch_allocator_->AllocateBytes(size_in_bytes); in MIOpenAllocatorCallback()
2872 auto allocated = scratch_allocator->AllocateBytes(scratch_memory_size); in DoPrepareForConvolution()
3403 auto allocated = scratch_allocator->AllocateBytes(scratch_memory_size); in GetMIOpenConvolveAlgorithmsFindMode()
4245 workspace_allocator->AllocateBytes(workspace_size_in_bytes); in DoPoolBackwardImpl()
4264 auto allocated = workspace_allocator->AllocateBytes(dest2_size); in DoPoolBackwardImpl()
4421 workspace_allocator->AllocateBytes(workspace_size_in_bytes); in DoNormalizeBackwardWithDimensions()
4446 auto allocated = workspace_allocator->AllocateBytes(dest2_size); in DoNormalizeBackwardWithDimensions()
|
| D | rocm_fft.cc | 332 auto allocated = scratch_allocator->AllocateBytes(scratch_size_bytes_); in UpdateScratchAllocator()
|