Searched refs:AllocateTemporaryArray (Results 1 – 5 of 5) sorted by relevance
36 stream->AllocateTemporaryArray<uint8>(byte_size)); in AllocateBytes()
150 AllocateTemporaryArray(uint64 element_count);2040 Stream::AllocateTemporaryArray(uint64 element_count) { in AllocateTemporaryArray() function
422 auto temp_status = stream->AllocateTemporaryArray<char>( in ExecuteOnStream()
2192 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasGemmBatchedInternal()2194 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasGemmBatchedInternal()2196 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasGemmBatchedInternal()
2782 stream->AllocateTemporaryArray<uint8>(backward_output_data.size()) in MaybeTransformLayout()