Searched refs:tf_dlm_tensor_ctx (Results 1 – 2 of 2) sorted by relevance
271 auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref); in TFE_HandleToDLPack() local272 tf_dlm_tensor_ctx->reference = tensor_ref; in TFE_HandleToDLPack()274 DLManagedTensor* dlm_tensor = &tf_dlm_tensor_ctx->tensor; in TFE_HandleToDLPack()275 dlm_tensor->manager_ctx = tf_dlm_tensor_ctx; in TFE_HandleToDLPack()283 std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape; in TFE_HandleToDLPack()284 std::vector<int64_t>* stride_arr = &tf_dlm_tensor_ctx->strides; in TFE_HandleToDLPack()
18 auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref);