Lines Matching refs:getTensor
62 NIMR.getTensor<int64_t>(0)[0] = 1; in TEST()
63 std::memcpy(NIMR.getTensor<int64_t>(1), in TEST()
66 std::memcpy(NIMR.getTensor<float>(2), in TEST()
69 ASSERT_EQ(NIMR.getTensor<int64_t>(0)[0], 1); in TEST()
70 ASSERT_EQ(NIMR.getTensor<int64_t>(1)[8], 9); in TEST()
71 ASSERT_EQ(NIMR.getTensor<float>(2)[1], 0.2f); in TEST()
80 *Evaluator->getTensor<int64_t>(0) = 1; in TEST()
81 *Evaluator->getTensor<int64_t>(1) = 2; in TEST()
83 EXPECT_EQ(*Evaluator->getTensor<int64_t>(0), 1); in TEST()
84 EXPECT_EQ(*Evaluator->getTensor<int64_t>(1), 2); in TEST()
94 *Evaluator->getTensor<int64_t>(0) = 1; in TEST()
95 *Evaluator->getTensor<int64_t>(1) = 2; in TEST()
96 *Evaluator->getTensor<int64_t>(2) = -3; in TEST()
98 EXPECT_EQ(*Evaluator->getTensor<int64_t>(0), 1); in TEST()
99 EXPECT_EQ(*Evaluator->getTensor<int64_t>(1), 2); in TEST()
100 EXPECT_EQ(*Evaluator->getTensor<int64_t>(2), -3); in TEST()
112 *Evaluator->getTensor<int64_t>(0) = 1; // a in TEST()
113 *Evaluator->getTensor<int64_t>(1) = 2; // c in TEST()
114 *Evaluator->getTensor<int64_t>(2) = -3; // b in TEST()
116 EXPECT_EQ(*Evaluator->getTensor<int64_t>(0), 1); in TEST()
117 EXPECT_EQ(*Evaluator->getTensor<int64_t>(1), 2); in TEST()
118 EXPECT_EQ(*Evaluator->getTensor<int64_t>(2), -3); in TEST()