Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for DLPack (0.15 sec)

  1. tensorflow/c/eager/dlpack.h

    namespace tensorflow {
    
    // PyCapsule name for DLPack Tensor
    const char* const kDlTensorCapsuleName = "dltensor";
    
    // Converts eager tensor handle to DLPack (DLManagedTensor*), and return the
    // void* for further PyCapsule construction.
    TF_CAPI_EXPORT extern void* TFE_HandleToDLPack(TFE_TensorHandle* h,
                                                   TF_Status* status);
    
    // Converts DLPack (DLManagedTensor*) to eager tensor handle.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Mar 28 08:41:24 UTC 2020
    - 1.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/dlpack_test.cc

    limitations under the License.
    ==============================================================================*/
    
    #include "tensorflow/c/eager/dlpack.h"
    
    #include <vector>
    
    #include "absl/strings/str_join.h"
    #include "include/dlpack/dlpack.h"  // from @dlpack
    #include "tensorflow/c/eager/c_api.h"
    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace {
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Fri Jun 30 03:04:46 UTC 2023
    - 4.4K bytes
    - Viewed (0)
Back to top