Searched defs:GetMemoryLimitInBytes (Results 1 – 7 of 7) sorted by relevance
68 int64_t GetMemoryLimitInBytes() override { return -1; } in GetMemoryLimitInBytes() function90 int64_t GetMemoryLimitInBytes() override { return -1; } in GetMemoryLimitInBytes() function
52 int64_t GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
52 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
681 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp734 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
369 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon980e2df00111::CudnnRnnAllocatorInTemp414 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon980e2df00111::CudnnRnnAllocatorInOutput453 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon980e2df00111::CudnnRNNSpaceAllocator
378 int64_t GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anon055adecb0111::CufftScratchAllocator
67 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anon042fc3bc0111::ScratchAllocator