Searched refs:rz_allocator (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 40 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator, in WrapRedzoneBestEffort() argument
45 auto output_rz_or = rz_allocator->AllocateBytes(buffer.size()); in WrapRedzoneBestEffort()
59 void CheckRedzones(const se::RedzoneAllocator& rz_allocator, in CheckRedzones() argument
65 rz_allocator.CheckRedzones(); in CheckRedzones()
|
| D | gpu_utils.h | 53 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator,
59 void CheckRedzones(const se::RedzoneAllocator& rz_allocator,
|
| D | conv_grad_input_ops.cc | 396 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()() local
400 WrapRedzoneBestEffort(&rz_allocator, in_backprop_ptr)); in operator ()()
438 CheckRedzones(rz_allocator, &result); in operator ()()
|
| D | conv_ops_3d.cc | 512 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in launch() local
515 WrapRedzoneBestEffort(&rz_allocator, output_ptr)); in launch()
559 CheckRedzones(rz_allocator, &result); in launch()
|
| D | conv_ops_fused_impl.h | 388 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter,
391 WrapRedzoneBestEffort(&rz_allocator, output_ptr));
422 CheckRedzones(rz_allocator, &result);
|
| D | conv_grad_filter_ops.cc | 990 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()() local
994 WrapRedzoneBestEffort(&rz_allocator, filter_backprop_ptr)); in operator ()()
1034 CheckRedzones(rz_allocator, &result); in operator ()()
|
| D | conv_ops.cc | 1008 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()() local
1011 WrapRedzoneBestEffort(&rz_allocator, output_ptr)); in operator ()()
1046 CheckRedzones(rz_allocator, &result); in operator ()()
|
| D | conv_grad_ops_3d.cc | 1430 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in Compute() local
1433 WrapRedzoneBestEffort(&rz_allocator, in_backprop_ptr)); in Compute()
1475 CheckRedzones(rz_allocator, &result); in Compute()
|