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
|