Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h92 port::Status SynchronousMemZero(DeviceMemoryBase *location,
Dhost_gpu_executor.cc87 port::Status HostExecutor::SynchronousMemZero(DeviceMemoryBase *location, in SynchronousMemZero() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h107 port::Status SynchronousMemZero(DeviceMemoryBase *location, in SynchronousMemZero() function
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.h237 port::Status SynchronousMemZero(DeviceMemoryBase *location,
855 bool ok = SynchronousMemZero(&result, sizeof(T)).ok(); in AllocateZeroed()
Dstream_executor_internal.h219 virtual port::Status SynchronousMemZero(DeviceMemoryBase *location,
Dstream_executor_pimpl.cc591 port::Status StreamExecutor::SynchronousMemZero(DeviceMemoryBase *location, in SynchronousMemZero() function in stream_executor::StreamExecutor
596 return implementation_->SynchronousMemZero(location, size); in SynchronousMemZero()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h212 Status SynchronousMemZero(DeviceMemoryBase* location, uint64 size) override { in SynchronousMemZero() function
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h122 port::Status SynchronousMemZero(DeviceMemoryBase* location,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc458 port::Status GpuExecutor::SynchronousMemZero(DeviceMemoryBase* location, in SynchronousMemZero() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc341 port::Status SynchronousMemZero(DeviceMemoryBase* location, in SynchronousMemZero() function in stream_executor::__anon00950a620111::CStreamExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc550 port::Status GpuExecutor::SynchronousMemZero(DeviceMemoryBase* location, in SynchronousMemZero() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc417 TF_RETURN_IF_ERROR(executor_->SynchronousMemZero( in InitScratchBuffers()