Searched defs:GetMemoryLimitInBytes (Results 1 – 9 of 9) sorted by relevance
29 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes() { return -1; } in GetMemoryLimitInBytes() function in stream_executor::OneTimeScratchAllocator
48 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
374 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anoncd6103410111::CufftScratchAllocator
692 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp745 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
368 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon787752e60111::CudnnRnnAllocatorInTemp413 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon787752e60111::CudnnRnnAllocatorInOutput452 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon787752e60111::CudnnRNNSpaceAllocator
53 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
34 int64 FftScratchAllocator::GetMemoryLimitInBytes() { in GetMemoryLimitInBytes() function in xla::gpu::FftScratchAllocator
52 int64 GetMemoryLimitInBytes() override { return scratch_.size(); } in GetMemoryLimitInBytes() function in xla::gpu::__anon66499ca80111::ScratchBufAllocator
62 int64 GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anonbaa4aab20111::ScratchAllocator