Searched refs:executionStep (Results 1 – 4 of 4) sorted by relevance
155 ASSERT_EQ(steps[1]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()158 ASSERT_EQ(steps[3]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()161 ASSERT_EQ(steps[5]->executionStep()->getDevice(), mTestDevice); in TEST_F()
1199 if (const auto* executionStep = step->tryExecutionStep()) { in TEST_P() local1200 devicesInPlan.insert(executionStep->getDevice().get()); in TEST_P()1206 if (const auto executionStep = steps[i]->tryExecutionStep()) { in TEST_P() local1208 << ModelStats(executionStep->getStepModel()) in TEST_P()1209 << ", device = " << executionStep->getDevice()->getName() in TEST_P()
1387 const ExecutionStep* step = logicalStep->executionStep(); in compare()2097 step->executionStep()->getDevice()->getName() == deviceName) { in getTransformedCacheTokenSingle()2099 token = step->executionStep()->forTest_getCacheToken(); in getTransformedCacheTokenSingle()2656 ASSERT_EQ(steps[1]->executionStep()->getDevice()->getName(), "V1_0"); in TEST_F()2657 ASSERT_EQ(steps[3]->executionStep()->getDevice()->getName(), "V1_0"); in TEST_F()2679 ASSERT_EQ(steps[1]->executionStep()->getDevice()->getName(), in TEST_F()2681 ASSERT_EQ(steps[3]->executionStep()->getDevice()->getName(), "V1_0"); in TEST_F()
355 ExecutionStep* executionStep() { return &std::get<ExecutionStep>(mStep); } in executionStep() function361 const ExecutionStep* executionStep() const { return &std::get<ExecutionStep>(mStep); } in executionStep() function