Searched defs:rz_allocator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 40 se::DeviceMemoryBase WrapRedzoneBestEffort(se::RedzoneAllocator* rz_allocator, in WrapRedzoneBestEffort()
59 void CheckRedzones(const se::RedzoneAllocator& rz_allocator, in CheckRedzones()
|
| D | conv_grad_input_ops.cc | 396 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()() local
|
| D | conv_ops_3d.cc | 512 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in launch() local
|
| D | conv_grad_filter_ops.cc | 990 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()() local
|
| D | conv_ops.cc | 1008 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in operator ()() local
|
| D | conv_grad_ops_3d.cc | 1430 se::RedzoneAllocator rz_allocator(stream, &tf_allocator_adapter, in Compute() local
|