Home
last modified time | relevance | path

Searched refs:SynchronousMemZero (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h108 bool SynchronousMemZero(DeviceMemoryBase *location, uint64 size) override { in SynchronousMemZero() function
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h92 bool SynchronousMemZero(DeviceMemoryBase *location, uint64 size) override;
Dhost_gpu_executor.cc55 bool HostExecutor::SynchronousMemZero(DeviceMemoryBase *location, uint64 size) { in SynchronousMemZero() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h249 bool SynchronousMemZero(DeviceMemoryBase *location,
836 bool ok = SynchronousMemZero(&result, sizeof(T)); in AllocateZeroed()
Dstream_executor_internal.h216 virtual bool SynchronousMemZero(DeviceMemoryBase *location, uint64 size) = 0;
Dstream_executor_pimpl.cc604 bool StreamExecutor::SynchronousMemZero(DeviceMemoryBase *location, in SynchronousMemZero() function in stream_executor::StreamExecutor
609 return implementation_->SynchronousMemZero(location, size); in SynchronousMemZero()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h122 bool SynchronousMemZero(DeviceMemoryBase* location, uint64 size) override;
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc460 bool GpuExecutor::SynchronousMemZero(DeviceMemoryBase* location, uint64 size) { in SynchronousMemZero() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc579 bool GpuExecutor::SynchronousMemZero(DeviceMemoryBase* location, uint64 size) { in SynchronousMemZero() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc359 bool ok = executor_->SynchronousMemZero( in InitScratchBuffers()