Lines Matching refs:ExecutionBuilder
112 static MeasureTiming measureTiming(const ExecutionBuilder* execution) { in measureTiming()
152 ExecutionBuilder::ExecutionBuilder(const CompilationBuilder* compilation) in ExecutionBuilder() function in android::nn::ExecutionBuilder
164 : ExecutionBuilder(compilation) { in SimpleExecutionBuilder()
169 : ExecutionBuilder(compilation) { in CompoundExecutionBuilder()
173 const ModelBuilder* ExecutionBuilder::getSourceModel(uint32_t index) const { in getSourceModel()
177 int ExecutionBuilder::setInput(uint32_t index, const ANeuralNetworksOperandType* type, in setInput()
211 int ExecutionBuilder::setInputFromMemory(uint32_t index, const ANeuralNetworksOperandType* type, in setInputFromMemory()
259 int ExecutionBuilder::setOutput(uint32_t index, const ANeuralNetworksOperandType* type, in setOutput()
292 int ExecutionBuilder::setOutputFromMemory(uint32_t index, const ANeuralNetworksOperandType* type, in setOutputFromMemory()
339 int ExecutionBuilder::setMeasureTiming(bool measure) { in setMeasureTiming()
356 int ExecutionBuilder::getDuration(int32_t durationCode, uint64_t* duration) const { in getDuration()
416 int ExecutionBuilder::setTimeoutDuration(uint64_t duration) { in setTimeoutDuration()
435 std::optional<uint64_t> ExecutionBuilder::getTimeoutDuration() const { in getTimeoutDuration()
439 int ExecutionBuilder::setLoopTimeout(uint64_t duration) { in setLoopTimeout()
454 int ExecutionBuilder::enableInputAndOutputPadding(bool enable) { in enableInputAndOutputPadding()
469 int ExecutionBuilder::setReusable(bool reusable) { in setReusable()
479 int ExecutionBuilder::getOutputOperandDimensions(uint32_t index, uint32_t* dimensions) { in getOutputOperandDimensions()
508 int ExecutionBuilder::getOutputOperandRank(uint32_t index, uint32_t* rank) { in getOutputOperandRank()
530 bool ExecutionBuilder::checkAndSetComputationState(const char* name) { in checkAndSetComputationState()
565 int ExecutionBuilder::getValidationResultCode() { in getValidationResultCode()
572 bool ExecutionBuilder::areOutputsFullySpecified() { in areOutputsFullySpecified()
588 int ExecutionBuilder::prepareForCompute(const char* name) { in prepareForCompute()
604 ExecutionBuilder* executionBuilder) { in cpuFallbackFull()
972 int ExecutionBuilder::computeFenced(const std::vector<int>& waitFor, in computeFenced()
1010 int ExecutionBuilder::compute(std::shared_ptr<ExecutionCallback>* synchronizationCallback, in compute()
1090 std::vector<OutputShape> ExecutionBuilder::getInitialOutputShapes() const { in getInitialOutputShapes()
1121 bool ExecutionBuilder::updateOutputShapes(ErrorStatus status, in updateOutputShapes()
1143 bool ExecutionBuilder::updateMemories() { in updateMemories()
1152 int ExecutionBuilder::finishComputation(int result, const std::vector<OutputShape>& outputShapes) { in finishComputation()
1313 StepExecutor::StepExecutor(ExecutionBuilder* executionBuilder, const ModelBuilder* model, in StepExecutor()