Searched defs:GetMemoryLimitInBytes (Results 1 – 9 of 9) sorted by relevance
29 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes() { return -1; } in GetMemoryLimitInBytes() function in stream_executor::OneTimeScratchAllocator
52 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
35 int64 FftScratchAllocator::GetMemoryLimitInBytes() { in GetMemoryLimitInBytes() function in xla::gpu::FftScratchAllocator
51 int64 GetMemoryLimitInBytes() override { return scratch_.size(); } in GetMemoryLimitInBytes() function in xla::gpu::__anon2f3164c70111::ScratchBufAllocator
57 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anona42065910111::ScratchAllocator
56 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
367 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anon82cc35e00111::CufftScratchAllocator
579 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp632 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
366 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon016bc0c50111::CudnnRnnAllocatorInTemp411 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon016bc0c50111::CudnnRnnAllocatorInOutput450 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon016bc0c50111::CudnnRNNPersistentSpaceAllocator