Searched refs:ThenMemZero (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | memset_thunk.cc | 29 stream->ThenMemZero(&dest_data, dest_data.size()); in ExecuteOnStream()
|
D | cudnn_conv_algorithm_picker.cc | 203 stream.ThenMemZero(&buffer, buffer.size()); in PickBestAlgorithm()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 1991 stream->ThenMemZero(workspace, workspace_size_in_bytes); in CreateRnnWorkspace() 2069 stream->ThenMemZero(&reserve_space, reserve_space_size_in_bytes); in DoRnnForwardImpl() 2172 stream->ThenMemZero(input_backprop_data, size_data * sizeof(float)); in DoRnnBackwardImpl() 2177 stream->ThenMemZero(input_h_backprop_data, size_data * sizeof(float)); in DoRnnBackwardImpl() 2182 stream->ThenMemZero(input_c_backprop_data, size_data * sizeof(float)); in DoRnnBackwardImpl() 2211 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl() 3059 stream->ThenMemZero(batch_mean, batch_mean->size()); in DoBatchNormalizationForwardImpl() 3060 stream->ThenMemZero(batch_var, batch_var->size()); in DoBatchNormalizationForwardImpl()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | bias_op.cc | 551 stream->ThenMemZero(&output_ptr, output->NumElements() * sizeof(T)); in Compute()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 1765 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl() 2785 stream->ThenMemZero(&scratch_memory, scratch_memory.size()); in DoConvolve() 2848 stream->ThenMemZero(&filter_data, filter_data.size()); in DoConvolve() 3253 stream->ThenMemZero(batch_mean, batch_mean->size()); in DoBatchNormalizationForwardImpl() 3254 stream->ThenMemZero(batch_var, batch_var->size()); in DoBatchNormalizationForwardImpl()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | stream.h | 1728 Stream &ThenMemZero(DeviceMemoryBase *location, uint64 size);
|
D | stream.cc | 4819 Stream &Stream::ThenMemZero(DeviceMemoryBase *location, uint64 size) { in ThenMemZero() function in stream_executor::Stream
|