HomeSort by relevance Sort by last modified time
    Searched refs:StepExecutor (Results 1 - 4 of 4) sorted by null

  /frameworks/ml/nn/runtime/
ExecutionBuilder.h 39 class StepExecutor;
67 friend class StepExecutor;
106 // class StepExecutor is used to execute a single "step" in a
110 class StepExecutor {
121 StepExecutor(const ExecutionBuilder* executionBuilder,
125 // Map inputs and outputs from ExecutionBuilder to StepExecutor,
130 // Map inputs and outputs from ExecutionBuilder to StepExecutor,
ExecutionPlan.h 38 class StepExecutor;
92 // Map inputs and outputs from ExecutionBuilder to StepExecutor.
93 void mapInputsAndOutputs(std::shared_ptr<StepExecutor> stepExecutor) const;
166 // *executor is set to point to a new StepExecutor corresponding
197 int next(std::shared_ptr<Controller> controller, std::shared_ptr<StepExecutor>* executor) const;
200 int fallback(std::shared_ptr<Controller> controller, std::shared_ptr<StepExecutor>* executor) const;
ExecutionBuilder.cpp 161 // Should be similar to StepExecutor::setInputOrOutputFromTemporaryMemory()
201 // Should be similar to StepExecutor::setInputOrOutputFromTemporaryMemory()
223 StepExecutor executor(executionBuilder, executionBuilder->getModel(),
248 std::shared_ptr<StepExecutor> executor;
274 std::shared_ptr<StepExecutor> executor;
275 VLOG(EXECUTION) << "looking for next StepExecutor";
384 // StepExecutor::startCompute() is designed as
387 // StepExecutor::startCompute() isn't actually asynchronous
398 StepExecutor executor(this, mModel, device->getInterface(),
409 StepExecutor executor(this, mModel
    [all...]
ExecutionPlan.cpp 269 void ExecutionStep::mapInputsAndOutputs(std::shared_ptr<StepExecutor> stepExecutor) const {
271 stepExecutor->mapInput(mInputIndexSubModelToFromModel[i], i);
274 stepExecutor->mapOutput(mOutputIndexSubModelToFromModel[i], i);
562 std::shared_ptr<StepExecutor>* executor) const {
583 std::shared_ptr<StepExecutor>* executor) const {
604 *executor = std::make_shared<StepExecutor>(
633 *executor = std::make_shared<StepExecutor>(
    [all...]

Completed in 93 milliseconds