Searched defs:c2_tensor (Results 1 – 2 of 2) sorted by relevance
9 caffe2::Tensor c2_tensor(caffe2::CPU); in TEST() local24 caffe2::Tensor c2_tensor = caffe2::empty({4, 4}, at::kLong); in TEST() local38 caffe2::Tensor c2_tensor = caffe2::empty({4, 4}, at::kLong); in TEST() local61 caffe2::Tensor c2_tensor(caffe2::CPU); in TEST() local86 caffe2::Tensor c2_tensor(caffe2::CPU); in TEST() local92 caffe2::Tensor c2_tensor(caffe2::CPU); in TEST() local99 caffe2::Tensor c2_tensor(caffe2::CPU); in TEST() local109 caffe2::Tensor c2_tensor = caffe2::empty({5, 5}, at::kFloat); in TEST() local230 caffe2::Tensor c2_tensor(at_tensor); in TEST() local262 caffe2::Tensor c2_tensor(at_tensor.contiguous()); in TEST() local[all …]
26 caffe2::Tensor c2_tensor(caffe2::CUDA); in TEST() local45 caffe2::Tensor c2_tensor = in TEST() local64 caffe2::Tensor c2_tensor = in TEST() local140 caffe2::Tensor c2_tensor(at_tensor); in TEST() local