Searched refs:RedzoneCheckStatus (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 46 using RedzoneCheckStatus = RedzoneAllocator::RedzoneCheckStatus; typedef
181 static port::StatusOr<RedzoneCheckStatus> CheckRedzoneHost( in CheckRedzoneHost()
198 return RedzoneCheckStatus(name, user_allocation.opaque(), i, pattern64, in CheckRedzoneHost()
205 return RedzoneCheckStatus(name, user_allocation.opaque(), i, in CheckRedzoneHost()
209 return RedzoneCheckStatus::OK(); in CheckRedzoneHost()
250 static port::StatusOr<RedzoneCheckStatus> CheckRedzonesForBuffer( in CheckRedzonesForBuffer()
282 TF_ASSIGN_OR_RETURN(RedzoneCheckStatus lhs_check, in CheckRedzonesForBuffer()
285 TF_ASSIGN_OR_RETURN(RedzoneCheckStatus rhs_check, in CheckRedzonesForBuffer()
299 return RedzoneCheckStatus::OK(); in CheckRedzonesForBuffer()
302 port::StatusOr<RedzoneCheckStatus> RedzoneAllocator::CheckRedzones() const { in CheckRedzones()
[all …]
|
| D | redzone_allocator.h | 63 struct RedzoneCheckStatus { struct
64 RedzoneCheckStatus() = default;
66 RedzoneCheckStatus(absl::string_view buffer_name, void* user_buffer_address, in RedzoneCheckStatus() argument
74 static RedzoneCheckStatus OK() { return {}; } in OK() argument
98 port::StatusOr<RedzoneCheckStatus> CheckRedzones() const;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 31 using RedzoneCheckStatus = RedzoneAllocator::RedzoneCheckStatus; typedef
33 static void EXPECT_REDZONE_OK(port::StatusOr<RedzoneCheckStatus> status) { in EXPECT_REDZONE_OK()
39 port::StatusOr<RedzoneCheckStatus> status) { in EXPECT_REDZONE_VIOLATION()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 225 using RedzoneCheckStatus = se::RedzoneAllocator::RedzoneCheckStatus; in CheckRedzones() typedef
226 TF_ASSIGN_OR_RETURN(RedzoneCheckStatus redzone_check, in CheckRedzones()
|
| D | gemm_algorithm_picker.cc | 161 se::RedzoneAllocator::RedzoneCheckStatus rz_check_status, in DoUncachedGemmAutotune()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 64 se::port::StatusOr<se::RedzoneAllocator::RedzoneCheckStatus> rz_status = in CheckRedzones()
|