Searched defs:GetMemoryLimitInBytes (Results 1 – 9 of 9) sorted by relevance
29 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes() { return -1; } in GetMemoryLimitInBytes() function in stream_executor::OneTimeScratchAllocator
53 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
56 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
367 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anon049b3f3c0111::CufftScratchAllocator
671 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp724 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
366 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInTemp411 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anond88b5f210111::CudnnRnnAllocatorInOutput450 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anond88b5f210111::CudnnRNNPersistentSpaceAllocator
35 int64 FftScratchAllocator::GetMemoryLimitInBytes() { in GetMemoryLimitInBytes() function in xla::gpu::FftScratchAllocator
52 int64 GetMemoryLimitInBytes() override { return scratch_.size(); } in GetMemoryLimitInBytes() function in xla::gpu::__anon53f8f4230111::ScratchBufAllocator
60 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anon6437d7ed0111::ScratchAllocator