Searched refs:simple_tensor (Results 1 – 3 of 3) sorted by relevance
163 SimpleITensor* simple_tensor() const { in simple_tensor() function441 p1->simple_tensor() == p2->simple_tensor()));
268 EXPECT_EQ(itensor->simple_tensor(), ptr->tensor()->simple_tensor()); in TEST()287 EXPECT_NE(nullptr, ptr->tensor()->simple_tensor()); in TEST()379 EXPECT_NE(nullptr, output.tensor()->simple_tensor()); in TEST_F()604 EXPECT_EQ(output_tensors[0]->simple_tensor(), in TEST_F()605 actual_output_1.tensor()->simple_tensor()); in TEST_F()610 EXPECT_EQ(output_tensors[1]->simple_tensor(), in TEST_F()611 actual_output_2.tensor()->simple_tensor()); in TEST_F()833 ITensorProxyPtr simple_tensor; in TEST_F() local834 TRT_TensorOrWeights tensor(simple_tensor); in TEST_F()872 ITensorProxyPtr simple_tensor; in TEST_F() local[all …]
891 Tensor simple_tensor(dtypes_[i], shape); in SnappyUncompress() local892 TensorBuffer* buffer = DMAHelper::buffer(&simple_tensor); in SnappyUncompress()895 simple_tensors->push_back(std::move(simple_tensor)); in SnappyUncompress()