Lines Matching defs:ModelRunner
114 NoInferenceModelRunner ModelRunner = setupModelRunner();
125 OverlapProblem, &ModelRunner,
131 ModelRunner.getTensor<int64_t>(1),
132 ModelRunner.getTensor<int64_t>(1) +
172 NoInferenceModelRunner ModelRunner = setupModelRunner();
176 OverlapProblem, &ModelRunner,
187 (size_t)ModelRunner.getTensor<int64_t>(0)[CurrentInstructionIndex],
237 NoInferenceModelRunner ModelRunner = setupModelRunner();
243 [](SlotIndex InputSlot) -> float { return 1.0f; }, nullptr, &ModelRunner,
245 ASSERT_FLOAT_EQ(ModelRunner.getTensor<float>(2)[1], 1.0f);
246 ASSERT_EQ(ModelRunner.getTensor<int64_t>(3)[0], 1);
254 NoInferenceModelRunner ModelRunner = setupModelRunner();
268 OverlapProblem, &ModelRunner,
280 ASSERT_FLOAT_EQ(ModelRunner.getTensor<float>(2)[MBBIndex],
282 ASSERT_EQ(ModelRunner.getTensor<int64_t>(3)[MBBIndex],
288 ASSERT_EQ(ModelRunner.getTensor<int64_t>(3)[MBBIndex],