Searched refs:ThenMemset32 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | replica_id_thunk.cc | 32 params.stream->ThenMemset32(&dest_addr, id, /*size=*/4); in ExecuteOnStream()
|
| D | memset_thunk.cc | 38 params.stream->ThenMemset32(&dest_data, value_, dest_data.size()); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 107 stream_->ThenMemset32(&lhs_redzone, pattern32, redzone_size_); in AllocateBytes()
111 stream_->ThenMemset32(&rhs_redzone_nonslop, pattern32, redzone_size_); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | debug_ops.h | 767 stream->ThenMemset32(&output_tensor_ptr, 0, 5 * sizeof(Tout));
794 stream->ThenMemset32(&output_tensor_ptr, 0, 11 * sizeof(Tout));
859 stream->ThenMemset32(&output_tensor_ptr, 0,
|
| D | check_numerics_op.cc | 229 stream->ThenMemset32(&abnormal_detected_ptr, 0, in ComputeAsync()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1775 Stream &ThenMemset32(DeviceMemoryBase *location, uint32 pattern, uint64 size);
|
| D | stream.cc | 4530 Stream &Stream::ThenMemset32(DeviceMemoryBase *location, uint32 pattern, in ThenMemset32() function in stream_executor::Stream
|