Searched refs:getTempsAsSubModelInputs (Results 1 - 3 of 3) sorted by relevance

/frameworks/ml/nn/runtime/test/
H A DTestPartitioning.cpp101 // getTempsAsSubModelInputs(), getOutputsAsSubModelInputs()).
107 // submodel inputs from getTempsAsSubModelInputs() in order,
823 ASSERT_EQ(stepsB[0]->getTempsAsSubModelInputs(),
849 ASSERT_EQ(stepsB[1]->getTempsAsSubModelInputs(),
915 ASSERT_EQ(step0->getTempsAsSubModelInputs(),
940 ASSERT_EQ(step1->getTempsAsSubModelInputs(),
965 ASSERT_EQ(step2->getTempsAsSubModelInputs(),
1064 ASSERT_EQ(steps[0]->getTempsAsSubModelInputs(),
1084 ASSERT_EQ(steps[1]->getTempsAsSubModelInputs(),
/frameworks/ml/nn/runtime/
H A DExecutionPlan.h61 const RemapVectorType& getTempsAsSubModelInputs() const { function in class:android::nn::ExecutionStep
H A DExecutionPlan.cpp280 for (const auto& input : step->getTempsAsSubModelInputs()) {
665 const auto& subModelInputs = step->getTempsAsSubModelInputs();
687 step->getModelInputs().size() + step->getTempsAsSubModelInputs().size();

Completed in 56 milliseconds