Searched refs:executionStep (Results 1 – 5 of 5) 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()
1397 const ExecutionStep* step = logicalStep->executionStep(); in compare()2106 step->executionStep()->getDevice()->getName() == deviceName) { in getTransformedCacheTokenSingle()2108 token = step->executionStep()->forTest_getCacheToken(); in getTransformedCacheTokenSingle()2670 ASSERT_EQ(steps[1]->executionStep()->getDevice()->getName(), "V1_0"); in TEST_F()2671 ASSERT_EQ(steps[3]->executionStep()->getDevice()->getName(), "V1_0"); in TEST_F()2693 ASSERT_EQ(steps[1]->executionStep()->getDevice()->getName(), in TEST_F()2695 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
398 mSteps[stepIndex]->executionStep()->recordTempAsStepModelOutput(sourceOperandIndex.second); in findTempsAsStepModelOutputs()606 executionStep()->dump(); in dump()789 logicalStep->executionStep()->getPreparedStepModel()) { in makeBursts()1442 return step->executionStep(); in createNewExecutionStep()