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()
1168 return step->executionStep()->getDevice(); in TEST_P()1193 if (const size_t count = step->executionStep() in TEST_P()1229 if (const auto* executionStep = step->tryExecutionStep()) { in TEST_P() local1230 devicesInPlan.insert(executionStep->getDevice().get()); in TEST_P()1236 if (const auto executionStep = steps[i]->tryExecutionStep()) { in TEST_P() local1238 << ModelStats(executionStep->getStepModel()) in TEST_P()1239 << ", device = " << executionStep->getDevice()->getName() in TEST_P()
1210 actual.emplace_back(step->executionStep()->getDevice()->getName()); in checkExecutionPlanSteps()1543 const ExecutionStep* step = logicalStep->executionStep(); in compare()2668 step->executionStep()->getDevice()->getName() == deviceName) { in getTransformedCacheTokenSingle()2670 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
628 mSteps[stepIndex]->executionStep()->declareModelOutputIsDownstreamInput( in findModelOutputsThatAreDownstreamInputs()652 mSteps[stepIndex]->executionStep()->recordTempAsStepModelOutput(sourceOperandIndex.second); in findTempsAsStepModelOutputs()869 executionStep()->dump(); in dump()1172 logicalStep->executionStep()->getPreparedStepModel()) { in makeBursts()1859 return step->executionStep(); in createNewExecutionStep()2033 const auto* step = mSteps[stepIndex]->executionStep(); in forEachStepRoleOfSourceOperand()