Home
last modified time | relevance | path

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

/honee/problems/torch/
H A Dsgs_model_torch.cpp94 double *input_tensor_ptr; in ModelInference_Torch() local
97 … PetscCallCXX(input_tensor_ptr = (double *)input_tensor.contiguous().to(torch::kCPU).data_ptr()); in ModelInference_Torch()
100 printf("%f\n", input_tensor_ptr[i]); in ModelInference_Torch()