Searched refs:GetMemoryLimitInBytes (Results 1 – 12 of 12) sorted by relevance
24 if (byte_size > GetMemoryLimitInBytes(stream)) { in AllocateBytes()29 byte_size, GetMemoryLimitInBytes(stream))); in AllocateBytes()
35 int64 FftScratchAllocator::GetMemoryLimitInBytes(se::Stream* stream) { in GetMemoryLimitInBytes() function in xla::gpu::FftScratchAllocator43 if (byte_size > GetMemoryLimitInBytes(stream)) { in AllocateBytes()48 byte_size, GetMemoryLimitInBytes(stream))); in AllocateBytes()
35 int64 GetMemoryLimitInBytes(se::Stream* stream) override { in GetMemoryLimitInBytes() function
43 int64 GetMemoryLimitInBytes(se::Stream* stream) override;
87 int64 GetMemoryLimitInBytes(se::Stream* /*stream*/) override { in GetMemoryLimitInBytes() function in xla::gpu::__anon2b86fa710111::ScratchBufAllocator
48 virtual int64 GetMemoryLimitInBytes(Stream* stream) = 0;69 int64 GetMemoryLimitInBytes(Stream* stream) override;
28 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes(Stream* stream) { in GetMemoryLimitInBytes() function in stream_executor::OneTimeScratchAllocator
44 int64 GetMemoryLimitInBytes(se::Stream* stream) override { in GetMemoryLimitInBytes() function
366 int64 GetMemoryLimitInBytes(Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anon454a70830111::CudnnRnnAllocatorInTemp412 int64 GetMemoryLimitInBytes(Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anon454a70830111::CudnnRnnAllocatorInOutput452 int64 GetMemoryLimitInBytes(Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anon454a70830111::CudnnRNNPersistentSpaceAllocator
316 int64 GetMemoryLimitInBytes(se::Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anona0c96f1e0111::CufftScratchAllocator
263 int64 GetMemoryLimitInBytes(Stream* stream) override { return -1; }
2355 ? std::max(scratch_allocator->GetMemoryLimitInBytes(stream), 0ll) in GetCudnnConvolutionForwardAlgorithm()2404 ? std::max(scratch_allocator->GetMemoryLimitInBytes(stream), 0ll) in GetCudnnConvolutionBackwardDataAlgorithm()2453 ? std::max(scratch_allocator->GetMemoryLimitInBytes(stream), 0ll) in GetCudnnConvolutionBackwardFilterAlgorithm()