Searched refs:AllocateTemporaryArray (Results 1 – 5 of 5) sorted by relevance
35 stream_->AllocateTemporaryArray<uint8>(byte_size)); in AllocateBytes()
164 AllocateTemporaryArray(uint64 element_count);2146 Stream::AllocateTemporaryArray(uint64 element_count) { in AllocateTemporaryArray() function
2412 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasInternalImpl()2414 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasInternalImpl()2416 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasInternalImpl()
2911 stream->AllocateTemporaryArray<uint8>(backward_output_data.size()) in MaybeTransformLayout()4137 wsp_mem = stream->AllocateTemporaryArray<uint8>(workspace_size) in DoPoolForward()
2024 SE_ASSIGN_OR_RETURN(*temp_memory, stream->AllocateTemporaryArray<MAPPED_T>( in AllocateStridedBuffer()