Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_algorithm_picker.cc61 class ScratchAllocator : public se::ScratchAllocator { class
63 ScratchAllocator(int device_ordinal, in ScratchAllocator() function in xla::gpu::__anon042fc3bc0111::ScratchAllocator
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dfft.h57 class ScratchAllocator; variable
H A Dblas.h60 class ScratchAllocator; variable
H A Dstream.h75 class ScratchAllocator; variable
H A Ddnn.cc126 bool /*use_fallback*/, ScratchAllocator* /*scratch_allocator*/, in GetConvolveRunners() argument
H A Ddnn.h54 class ScratchAllocator; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc4720 ScratchAllocator* /*scratch_allocator*/, in GetConvolveRunners() argument