Home
last modified time | relevance | path

Searched defs:c2_tensor_a (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dtensor_interop_test.cpp151 auto* c2_tensor_a = BlobSetTensor(workspace.CreateBlob("a"), caffe2::Tensor(at_tensor_a)); in TEST() local
190 auto* c2_tensor_a = BlobSetTensor(workspace.CreateBlob("a"), caffe2::Tensor(at_tensor_a)); in TEST() local
217 caffe2::Tensor c2_tensor_a(at_tensor_a); in TEST() local
H A Dcuda_tensor_interop_test.cpp86 auto* c2_tensor_a = BlobSetTensor(workspace.CreateBlob("a"), caffe2::Tensor(at_tensor_a)); in TEST() local
126 caffe2::Tensor c2_tensor_a(at_tensor_a); in TEST() local