Home
last modified time | relevance | path

Searched refs:rz_allocator (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.cc41 size_t scratch_size_limit, const se::RedzoneAllocator& rz_allocator) { in AutotuneConvImpl() argument
87 CheckRedzones(rz_allocator, &result); in AutotuneConvImpl()
129 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in AutotuneFusedConv() local
132 WrapRedzoneBestEffort(&rz_allocator, output_ptr)); in AutotuneFusedConv()
156 scratch_size_limit, rz_allocator)); in AutotuneFusedConv()
197 launch_func, scratch_size_limit, rz_allocator)); in AutotuneFusedConv()
292 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in AutotuneUnfusedConv() local
301 WrapRedzoneBestEffort(&rz_allocator, output_ptr)); in AutotuneUnfusedConv()
305 WrapRedzoneBestEffort(&rz_allocator, input_ptr)); in AutotuneUnfusedConv()
309 WrapRedzoneBestEffort(&rz_allocator, filter_ptr)); in AutotuneUnfusedConv()
[all …]
H A Dgpu_utils.cc41 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator, in WrapRedzoneBestEffort() argument
46 auto output_rz_or = rz_allocator->AllocateBytes(buffer.size()); in WrapRedzoneBestEffort()
60 void CheckRedzones(const se::RedzoneAllocator& rz_allocator, in CheckRedzones() argument
66 rz_allocator.CheckRedzones(); in CheckRedzones()
H A Dgpu_utils.h63 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator,
69 void CheckRedzones(const se::RedzoneAllocator& rz_allocator,