Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dscratch_allocator.h68 int64_t GetMemoryLimitInBytes() override { return -1; } in GetMemoryLimitInBytes() function
90 int64_t GetMemoryLimitInBytes() override { return -1; } in GetMemoryLimitInBytes() function
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dredzone_allocator.h52 int64_t GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h52 int64 GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function
Dfft_ops.cc378 int64_t GetMemoryLimitInBytes() override { return memory_limit_; } in GetMemoryLimitInBytes() function in tensorflow::__anon4f5ed94b0111::CufftScratchAllocator
Dfused_batch_norm_op.cc681 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInTemp
734 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::functor::CudnnBatchNormAllocatorInOutput
Dcudnn_rnn_ops.cc369 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anonf4c408700111::CudnnRnnAllocatorInTemp
414 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anonf4c408700111::CudnnRnnAllocatorInOutput
453 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in tensorflow::__anonf4c408700111::CudnnRNNSpaceAllocator
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc67 int64_t GetMemoryLimitInBytes() override { in GetMemoryLimitInBytes() function in xla::gpu::__anond2bd1e3c0111::ScratchAllocator