Searched refs:ThenMemset32 (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | memset_thunk.cc | 38 stream->ThenMemset32(&dest_data, value_, dest_data.size()); in ExecuteOnStream()
|
| D | cudnn_conv_algorithm_picker.cc | 193 stream.ThenMemset32(&buffer, bits, aligned_size); in PickBestAlgorithm()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | check_numerics_op.cc | 149 stream->ThenMemset32(&abnormal_detected_ptr, 0, in ComputeAsync()
|
| D | depthwise_conv_op_gpu.h | 1789 stream->ThenMemset32(&filter_bp_ptr, 0, num_filter_backprop * sizeof(T));
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1733 Stream &ThenMemset32(DeviceMemoryBase *location, uint32 pattern, uint64 size);
|
| D | stream.cc | 4831 Stream &Stream::ThenMemset32(DeviceMemoryBase *location, uint32 pattern, in ThenMemset32() function in stream_executor::Stream
|