mahout icon indicating copy to clipboard operation
mahout copied to clipboard

[QDP] GPU TF tensor support with dlpack and device placement

Open rich7420 opened this issue 4 weeks ago • 0 comments

Summary

This issue asks GPU tensor support using dlpack protocol (TF 2.5+) and handles TensorFlow's device placement system.

  • extend encode_tf_tensor() to handle GPU tensors
  • add DLPack extraction (TF 2.5+)
  • handle TensorFlow device placement
  • Handle TensorFlow-specific scenarios
  • add stream synchronization
  • some tests

If anything important is missing, please leave a comment.

rich7420 avatar Dec 12 '25 12:12 rich7420