Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

Module: tf.experimental.dlpack

Public API for tf.experimental.dlpack namespace.

Functions

from_dlpack(...): Returns the Tensorflow eager tensor.

to_dlpack(...): Returns the dlpack capsule representing the tensor.