HomeSort by relevance Sort by last modified time
    Searched defs:GetMemoryLimitInBytes (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/stream_executor/
scratch_allocator.cc 29 int64 OneTimeScratchAllocator::GetMemoryLimitInBytes(Stream* stream) {
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
fft_thunk.cc 47 int64 FftScratchAllocator::GetMemoryLimitInBytes(se::Stream* stream) {
55 if (byte_size > GetMemoryLimitInBytes(stream)) {
60 byte_size, GetMemoryLimitInBytes(stream)));

Completed in 152 milliseconds