Searched refs:ThenMemZero (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | memset_thunk.cc | 26 params.stream->ThenMemZero(&dest_data, dest_data.size()); in ExecuteOnStream()
|
| D | nccl_collective_permute_thunk.cc | 236 stream.ThenMemZero(&dest_addr, dest_addr.size()); in RunCollectivePermute()
|
| D | buffer_comparator.cc | 621 stream->ThenMemZero(out_param.ptr(), sizeof(uint64_t)); in DeviceCompare()
|
| D | gpu_conv_algorithm_picker.cc | 808 stream->ThenMemZero(&buffer, buffer.size()); in PickBestAlgorithmNoCacheRocm()
|
| D | jitrt_custom_calls.cc | 1198 stream->ThenMemZero(&dst_data, dst.size_in_bytes); in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 106 .ThenMemZero(&redzone_at_offset, 1); in TEST()
|
| D | cuda_dnn.cc | 2230 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl()
2329 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl()
5541 stream->ThenMemZero(batch_mean, batch_mean->size()); in DoBatchNormalizationForwardImpl()
5542 stream->ThenMemZero(batch_var, batch_var->size()); in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op_gpu.cu.cc | 241 ->ThenMemZero(&elements_per_row_gpu_memory, in operator ()()
252 if (!stream->ThenMemZero(&rows_are_not_ordered_gpu_memory, sizeof(int)) in operator ()()
|
| D | bias_op.cc | 460 stream->ThenMemZero(&output_ptr, output->NumElements() * sizeof(T)); in Compute()
|
| D | debug_ops.h | 741 stream->ThenMemZero(&output_tensor_ptr, 2 * sizeof(Tout));
|
| D | list_kernels.h | 84 auto result = stream->ThenMemZero(&ptr, tensor.TotalBytes()).ok(); in SetZero()
|
| D | depthwise_conv_op_gpu.h | 1739 stream->ThenMemZero(&filter_bp_ptr, num_filter_backprop * sizeof(T));
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| D | redzone_allocator.cc | 325 stream_->ThenMemZero(out_param.ptr(), sizeof(uint64_t)); in CheckRedzones()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2126 stream->ThenMemZero(workspace, workspace_size_in_bytes); in CreateRnnWorkspace()
2204 stream->ThenMemZero(&reserve_space, reserve_space_size_in_bytes); in DoRnnForwardImpl()
2308 stream->ThenMemZero(input_backprop_data, size_data * type_size); in DoRnnBackwardImpl()
2313 stream->ThenMemZero(input_h_backprop_data, size_data * type_size); in DoRnnBackwardImpl()
2318 stream->ThenMemZero(input_c_backprop_data, size_data * type_size); in DoRnnBackwardImpl()
2347 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 829 Stream& stream_ref = stream.ThenMemZero(&device_data, size); in TEST_F()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 647 stream->ThenMemZero(&dev_workspace_ptr, work_size_in_bytes); in HeevdImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.h | 1216 Stream &ThenMemZero(DeviceMemoryBase *location, uint64_t size);
|
| D | stream.cc | 2015 Stream &Stream::ThenMemZero(DeviceMemoryBase *location, uint64_t size) { in ThenMemZero() function in stream_executor::Stream
|