Searched refs:executionStep (Results 1 – 5 of 5) sorted by relevance
154 ASSERT_EQ(steps[1]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()157 ASSERT_EQ(steps[3]->executionStep()->getDevice(), DeviceManager::getCpuDevice()); in TEST_F()160 ASSERT_EQ(steps[5]->executionStep()->getDevice(), mTestDevice); in TEST_F()
1152 return step->executionStep()->getDevice(); in TEST_P()1177 if (const size_t count = step->executionStep() in TEST_P()1213 if (const auto* executionStep = step->tryExecutionStep()) { in TEST_P() local1214 devicesInPlan.insert(executionStep->getDevice().get()); in TEST_P()1220 if (const auto executionStep = steps[i]->tryExecutionStep()) { in TEST_P() local1222 << ModelStats(executionStep->getStepModel()) in TEST_P()1223 << ", device = " << executionStep->getDevice()->getName() in TEST_P()
1211 actual.emplace_back(step->executionStep()->getDevice()->getName()); in checkExecutionPlanSteps()1544 const ExecutionStep* step = logicalStep->executionStep(); in compare()2669 step->executionStep()->getDevice()->getName() == deviceName) { in getTransformedCacheTokenSingle()2671 token = step->executionStep()->forTest_getCacheToken(); in getTransformedCacheTokenSingle()
504 ExecutionStep* executionStep() { return &std::get<ExecutionStep>(mStep); } in executionStep() function510 const ExecutionStep* executionStep() const { return &std::get<ExecutionStep>(mStep); } in executionStep() function
668 mSteps[stepIndex]->executionStep()->declareModelOutputIsDownstreamInput( in findModelOutputsThatAreDownstreamInputs()692 mSteps[stepIndex]->executionStep()->recordTempAsStepModelOutput(sourceOperandIndex.second); in findTempsAsStepModelOutputs()909 executionStep()->dump(); in dump()1224 logicalStep->executionStep()->getPreparedStepModel()) { in makeBursts()1911 return step->executionStep(); in createNewExecutionStep()2085 const auto* step = mSteps[stepIndex]->executionStep(); in forEachStepRoleOfSourceOperand()