Searched refs:out_param (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 218 const DeviceMemory<uint64>& out_param, in RunRedzoneChecker() argument
230 redzone.size(), out_param); in RunRedzoneChecker()
252 const DeviceMemory<uint64>& out_param, in CheckRedzonesForBuffer() argument
272 RunRedzoneChecker(stream, lhs_redzone, redzone_pattern, out_param, in CheckRedzonesForBuffer()
274 RunRedzoneChecker(stream, rhs_redzone, redzone_pattern, out_param, in CheckRedzonesForBuffer()
277 CHECK_EQ(out_param.size(), sizeof(result)); in CheckRedzonesForBuffer()
278 stream->ThenMemcpy(&result, out_param, sizeof(result)); in CheckRedzonesForBuffer()
321 ScopedDeviceMemory<uint64> out_param = in CheckRedzones() local
323 stream_->ThenMemZero(out_param.ptr(), sizeof(uint64)); in CheckRedzones()
334 CheckRedzonesForBuffer(stream_, *buf_and_size.first, out_param.cref(), in CheckRedzones()
|
| /external/mesa3d/src/vulkan/overlay-layer/ |
| D | overlay_params.c | 116 parse_string(const char *s, char *out_param, char *out_value) in parse_string() argument
120 for (; !is_delimiter(*s); s++, out_param++, i++) in parse_string()
121 *out_param = *s; in parse_string()
123 *out_param = 0; in parse_string()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator.cc | 569 se::ScopedDeviceMemory<uint64> out_param = in DeviceCompare() local
572 stream->ThenMemZero(out_param.ptr(), sizeof(uint64)); in DeviceCompare()
627 buffer_size, out_param.cref()); in DeviceCompare()
630 CHECK_EQ(out_param->size(), sizeof(result)); in DeviceCompare()
631 stream->ThenMemcpy(&result, *out_param, sizeof(result)); in DeviceCompare()
|