Searched refs:CheckRedzones (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 68 EXPECT_REDZONE_OK(allocator.CheckRedzones()); in TEST()
105 { EXPECT_REDZONE_OK(allocator.CheckRedzones()); } in TEST()
108 EXPECT_REDZONE_VIOLATION(allocator.CheckRedzones()); in TEST()
111 EXPECT_REDZONE_OK(allocator.CheckRedzones()); in TEST()
141 EXPECT_REDZONE_OK(allocator.CheckRedzones()); in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.h | 98 port::StatusOr<RedzoneCheckStatus> CheckRedzones() const;
|
| D | redzone_allocator.cc | 302 port::StatusOr<RedzoneCheckStatus> RedzoneAllocator::CheckRedzones() const { in CheckRedzones() function in stream_executor::RedzoneAllocator
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 59 void CheckRedzones(const se::RedzoneAllocator& rz_allocator, in CheckRedzones() function
65 rz_allocator.CheckRedzones(); in CheckRedzones()
|
| D | gpu_utils.h | 59 void CheckRedzones(const se::RedzoneAllocator& rz_allocator,
|
| D | conv_grad_input_ops.cc | 437 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
438 CheckRedzones(rz_allocator, &result); in operator ()()
|
| D | conv_ops_3d.cc | 558 CheckRedzones(rz_scratch_allocator, &result); in launch()
559 CheckRedzones(rz_allocator, &result); in launch()
|
| D | conv_ops_fused_impl.h | 421 CheckRedzones(rz_scratch_allocator, &result);
422 CheckRedzones(rz_allocator, &result);
|
| D | conv_grad_filter_ops.cc | 1033 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
1034 CheckRedzones(rz_allocator, &result); in operator ()()
|
| D | conv_ops.cc | 1045 CheckRedzones(rz_scratch_allocator, &result); in operator ()()
1046 CheckRedzones(rz_allocator, &result); in operator ()()
|
| D | conv_grad_ops_3d.cc | 1474 CheckRedzones(rz_scratch_allocator, &result); in Compute()
1475 CheckRedzones(rz_allocator, &result); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 219 StatusOr<bool> CheckRedzones(const se::RedzoneAllocator& allocator, in CheckRedzones() function
227 allocator.CheckRedzones()); in CheckRedzones()
502 CheckRedzones(input_output_allocator, stream, in PickBestAlgorithmNoCacheCuda()
507 CheckRedzones(scratch_allocator, stream, "scratch", instr, &result)); in PickBestAlgorithmNoCacheCuda()
|
| D | gemm_algorithm_picker.cc | 162 input_output_allocator.CheckRedzones()); in DoUncachedGemmAutotune()
|