Home
last modified time | relevance | path

Searched refs:scratch_allocator_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.h54 scratch_allocator_(nullptr) {} in CUDAFftPlan()
83 ScratchAllocator* GetScratchAllocator() const { return scratch_allocator_; } in GetScratchAllocator()
95 ScratchAllocator* scratch_allocator_; variable
Dcuda_fft.cc86 scratch_allocator_ = scratch_allocator; in Initialize()
248 scratch_allocator_ = scratch_allocator; in UpdateScratchAllocator()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.h94 ScratchAllocator *GetScratchAllocator() const { return scratch_allocator_; } in GetScratchAllocator()
98 ScratchAllocator *scratch_allocator_; variable
Drocm_fft.cc166 scratch_allocator_ = scratch_allocator; in Initialize()
332 scratch_allocator_ = scratch_allocator; in UpdateScratchAllocator()
Drocm_dnn.cc2876 : scratch_allocator_(scratch_allocator), stream_(stream) {} in MIOpenAllocatorContext()
2878 ScratchAllocator* scratch_allocator_; member
2884 auto allocated = mac->scratch_allocator_->AllocateBytes(size_in_bytes); in MIOpenAllocatorCallback()