Home
last modified time | relevance | path

Searched refs:ThenMemZero (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmemset_thunk.cc29 stream->ThenMemZero(&dest_data, dest_data.size()); in ExecuteOnStream()
Dcudnn_conv_algorithm_picker.cc203 stream.ThenMemZero(&buffer, buffer.size()); in PickBestAlgorithm()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1991 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/
Dbias_op.cc551 stream->ThenMemZero(&output_ptr, output->NumElements() * sizeof(T)); in Compute()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1765 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/
Dstream.h1728 Stream &ThenMemZero(DeviceMemoryBase *location, uint64 size);
Dstream.cc4819 Stream &Stream::ThenMemZero(DeviceMemoryBase *location, uint64 size) { in ThenMemZero() function in stream_executor::Stream