Home
last modified time | relevance | path

Searched refs:MakeFromByteSize (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddevice_memory.h141 static DeviceMemory<ElemT> MakeFromByteSize(void *opaque, uint64 bytes) { in MakeFromByteSize() function
Dstream_executor_pimpl.h767 return DeviceMemory<T>::MakeFromByteSize(opaque, bytes); in AllocateArray()
835 DeviceMemory<T> result = DeviceMemory<T>::MakeFromByteSize(opaque, sizeof(T)); in AllocateZeroed()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc294 return DeviceMemory<T>::MakeFromByteSize( in AsDeviceMemory()
301 return DeviceMemory<T>::MakeFromByteSize( in AsDeviceMemory()
308 return DeviceMemory<U>::MakeFromByteSize( in CastDeviceMemory()
385 return DeviceMemory<uint8>::MakeFromByteSize( in AllocateBytes()
429 auto memory_uint8 = DeviceMemory<uint8>::MakeFromByteSize( in AllocateBytes()
Dbatch_matmul_op_impl.h273 DeviceMemoryBytes::MakeFromByteSize(nullptr, 0));
279 DeviceMemoryBytes::MakeFromByteSize(
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3628 a[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()
3636 c[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3388 a[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()
3396 c[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()