Searched refs:RedzoneAllocator (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 31 using RedzoneCheckStatus = RedzoneAllocator::RedzoneCheckStatus;
61 RedzoneAllocator allocator( in TEST()
63 /*memory_limit=*/RedzoneAllocator::kDefaultMemoryLimit, in TEST()
135 RedzoneAllocator allocator( in TEST()
137 /*memory_limit=*/RedzoneAllocator::kDefaultMemoryLimit, in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 46 using RedzoneCheckStatus = RedzoneAllocator::RedzoneCheckStatus;
48 RedzoneAllocator::RedzoneAllocator(Stream* stream, in RedzoneAllocator() function in stream_executor::RedzoneAllocator
63 port::StatusOr<DeviceMemory<uint8>> RedzoneAllocator::AllocateBytes( in AllocateBytes()
302 port::StatusOr<RedzoneCheckStatus> RedzoneAllocator::CheckRedzones() const { in CheckRedzones()
|
| D | redzone_allocator.h | 40 class RedzoneAllocator : public ScratchAllocator {
46 RedzoneAllocator(Stream* stream, DeviceMemoryAllocator* memory_allocator,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.h | 33 class RedzoneAllocator; variable
53 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator,
59 void CheckRedzones(const se::RedzoneAllocator& rz_allocator,
|
| D | gpu_utils.cc | 40 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator, in WrapRedzoneBestEffort()
59 void CheckRedzones(const se::RedzoneAllocator& rz_allocator, in CheckRedzones()
64 se::port::StatusOr<se::RedzoneAllocator::RedzoneCheckStatus> rz_status = in CheckRedzones()
|
| D | conv_grad_input_ops.cc | 396 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()()
412 se::RedzoneAllocator rz_scratch_allocator( in operator ()()
|
| D | conv_ops_3d.cc | 512 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in launch()
532 se::RedzoneAllocator rz_scratch_allocator( in launch()
|
| D | conv_ops_fused_impl.h | 388 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter,
396 se::RedzoneAllocator rz_scratch_allocator(
|
| D | conv_grad_filter_ops.cc | 990 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()()
1006 se::RedzoneAllocator rz_scratch_allocator( in operator ()()
|
| D | conv_ops.cc | 1008 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()()
1017 se::RedzoneAllocator rz_scratch_allocator( in operator ()()
|
| D | conv_grad_ops_3d.cc | 1430 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in Compute()
1446 se::RedzoneAllocator rz_scratch_allocator( in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_algorithm_picker.cc | 70 se::RedzoneAllocator input_output_allocator( in DoUncachedGemmAutotune()
161 se::RedzoneAllocator::RedzoneCheckStatus rz_check_status, in DoUncachedGemmAutotune()
|
| D | gpu_conv_algorithm_picker.cc | 219 StatusOr<bool> CheckRedzones(const se::RedzoneAllocator& allocator, in CheckRedzones()
225 using RedzoneCheckStatus = se::RedzoneAllocator::RedzoneCheckStatus; in CheckRedzones()
403 se::RedzoneAllocator input_output_allocator( in PickBestAlgorithmNoCacheCuda()
463 se::RedzoneAllocator scratch_allocator( in PickBestAlgorithmNoCacheCuda()
|