Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddevice_memory.h146 static DeviceMemory<ElemT> MakeFromByteSize(void *opaque, uint64 bytes) { in MakeFromByteSize() function
/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()
384 return DeviceMemory<uint8>::MakeFromByteSize( in AllocateBytes()
427 auto memory_uint8 = DeviceMemory<uint8>::MakeFromByteSize( in AllocateBytes()
Dmatmul_op_impl.h305 DeviceMemoryBytes::MakeFromByteSize(nullptr, 0));
311 DeviceMemoryBytes::MakeFromByteSize(
Dfused_batch_norm_op.cc655 return DeviceMemory<U>::MakeFromByteSize( in CastDeviceMemory()
689 return DeviceMemory<uint8>::MakeFromByteSize( in AllocateBytes()
742 auto memory_uint8 = DeviceMemory<uint8>::MakeFromByteSize( in AllocateBytes()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4161 a[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()
4169 c[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3914 a[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()
3922 c[i] = DeviceMemory<float>::MakeFromByteSize( in DoMatMul()