Searched refs:c2_tensor (Results 1 – 2 of 2) sorted by relevance
9 caffe2::Tensor c2_tensor(caffe2::CPU); in TEST() local10 c2_tensor.Resize(4, 4); in TEST()11 auto data = c2_tensor.mutable_data<int64_t>(); in TEST()15 at::Tensor at_tensor(c2_tensor); in TEST()24 caffe2::Tensor c2_tensor = caffe2::empty({4, 4}, at::kLong); in TEST() local25 auto data = c2_tensor.mutable_data<int64_t>(); in TEST()29 at::Tensor at_tensor(c2_tensor); in TEST()38 caffe2::Tensor c2_tensor = caffe2::empty({4, 4}, at::kLong); in TEST() local44 c2_tensor.ShareExternalPointer(buf, 16 * sizeof(int64_t)); in TEST()48 at::Tensor at_tensor(c2_tensor); in TEST()[all …]
26 caffe2::Tensor c2_tensor(caffe2::CUDA); in TEST() local27 c2_tensor.Resize(4, 4); in TEST()28 auto data = c2_tensor.mutable_data<int64_t>(); in TEST()33 at::Tensor at_tensor(c2_tensor); in TEST()45 caffe2::Tensor c2_tensor = in TEST() local47 auto data = c2_tensor.mutable_data<int64_t>(); in TEST()52 at::Tensor at_tensor(c2_tensor); in TEST()64 caffe2::Tensor c2_tensor = in TEST() local66 auto data = c2_tensor.mutable_data<int64_t>(); in TEST()71 at::Tensor at_tensor(c2_tensor); in TEST()[all …]