Home
last modified time | relevance | path

Searched defs:GetMemoryLimitInBytes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dscratch_allocator.cc28 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes(Stream* stream) { in GetMemoryLimitInBytes() function in stream_executor::OneTimeScratchAllocator
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dscratch_allocator.h35 int64 GetMemoryLimitInBytes(se::Stream* stream) override { in GetMemoryLimitInBytes() function
Dfft_thunk.cc35 int64 FftScratchAllocator::GetMemoryLimitInBytes(se::Stream* stream) { in GetMemoryLimitInBytes() function in xla::gpu::FftScratchAllocator
Dcudnn_conv_runner.cc87 int64 GetMemoryLimitInBytes(se::Stream* /*stream*/) override { in GetMemoryLimitInBytes() function in xla::gpu::__anon47e1e5a80111::ScratchBufAllocator
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h44 int64 GetMemoryLimitInBytes(se::Stream* stream) override { in GetMemoryLimitInBytes() function
Dfft_ops.cc316 int64 GetMemoryLimitInBytes(se::Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anonbbe6ef150111::CufftScratchAllocator
Dcudnn_rnn_ops.cc366 int64 GetMemoryLimitInBytes(Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInTemp
412 int64 GetMemoryLimitInBytes(Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anonebbbcdba0111::CudnnRnnAllocatorInOutput
452 int64 GetMemoryLimitInBytes(Stream* stream) override { in GetMemoryLimitInBytes() function in tensorflow::__anonebbbcdba0111::CudnnRNNPersistentSpaceAllocator