Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
H A Dredzone_allocator.cc255 const ComparisonKernelT& comparison_kernel, int64_t user_allocation_size, in CheckRedzonesForBuffer() argument
259 RoundUpToNearest<int64_t>(user_allocation_size, kRhsRedzoneAlign) - in CheckRedzonesForBuffer()
260 user_allocation_size; in CheckRedzonesForBuffer()
261 CHECK_EQ(memory.size(), user_allocation_size + rhs_slop + 2 * redzone_size); in CheckRedzonesForBuffer()
269 /*element_count=*/user_allocation_size); in CheckRedzonesForBuffer()
271 executor->GetSubBuffer(&buffer_uint8, redzone_size + user_allocation_size, in CheckRedzonesForBuffer()