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
378 int64_t GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anon686d0f5a0111::CufftScratchAllocator
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::__anon6bcc4dbf0111::CudnnRnnAllocatorInTemp414 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon6bcc4dbf0111::CudnnRnnAllocatorInOutput453 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anon6bcc4dbf0111::CudnnRNNSpaceAllocator
67 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anon8ba6008b0111::ScratchAllocator