Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/test/
Dcuda_tensor_interop_test.cpp84 auto at_tensor_c = at::ones({5, 5}, at::dtype(at::kFloat).device(at::kCUDA)); in TEST() local
93 caffe2::Tensor c2_tensor_from_aten(at_tensor_c); in TEST()
Dtensor_interop_test.cpp148 auto at_tensor_c = at::ones({5, 5}, at::dtype(at::kFloat)); in TEST() local
157 caffe2::Tensor c2_tensor_from_aten(at_tensor_c); in TEST()