Searched refs:ThenMemZero (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | memset_thunk.cc | 29 params.stream->ThenMemZero(&dest_data, dest_data.size()); in ExecuteOnStream()
|
| D | collective_permute_thunk.cc | 269 params.stream->ThenMemZero(&dest_addr, dest_addr.size()); in ExecuteOnStream()
|
| D | buffer_comparator.cc | 572 stream->ThenMemZero(out_param.ptr(), sizeof(uint64)); in DeviceCompare()
|
| D | gpu_conv_algorithm_picker.cc | 669 stream->ThenMemZero(&buffer, buffer.size()); in PickBestAlgorithmNoCacheRocm()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 107 .ThenMemZero(&redzone_at_offset, 1); in TEST()
|
| D | cuda_dnn.cc | 1986 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl()
3579 stream->ThenMemZero(batch_mean, batch_mean->size()); in DoBatchNormalizationForwardImpl()
3580 stream->ThenMemZero(batch_var, batch_var->size()); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 323 stream_->ThenMemZero(out_param.ptr(), sizeof(uint64)); in CheckRedzones()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 519 stream->ThenMemZero(&output_ptr, output->NumElements() * sizeof(T)); in Compute()
|
| D | debug_ops.h | 740 stream->ThenMemZero(&output_tensor_ptr, 2 * sizeof(Tout));
|
| D | depthwise_conv_op_gpu.h | 1766 stream->ThenMemZero(&filter_bp_ptr, num_filter_backprop * sizeof(T));
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2072 stream->ThenMemZero(workspace, workspace_size_in_bytes); in CreateRnnWorkspace()
2150 stream->ThenMemZero(&reserve_space, reserve_space_size_in_bytes); in DoRnnForwardImpl()
2254 stream->ThenMemZero(input_backprop_data, size_data * type_size); in DoRnnBackwardImpl()
2259 stream->ThenMemZero(input_h_backprop_data, size_data * type_size); in DoRnnBackwardImpl()
2264 stream->ThenMemZero(input_c_backprop_data, size_data * type_size); in DoRnnBackwardImpl()
2293 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.h | 1770 Stream &ThenMemZero(DeviceMemoryBase *location, uint64 size);
|
| D | stream.cc | 4523 Stream &Stream::ThenMemZero(DeviceMemoryBase *location, uint64 size) { in ThenMemZero() function in stream_executor::Stream
|