Home
last modified time | relevance | path

Searched refs:isOmittedOutput (Results 1 – 6 of 6) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DSqueeze.cpp37 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
DTranspose.cpp77 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
DUnidirectionalSequenceLSTM.cpp272 NN_RET_CHECK(!context->isOmittedOutput(kOutputStateOutTensor)); in prepare()
273 NN_RET_CHECK(!context->isOmittedOutput(kCellStateOutTensor)); in prepare()
DUnidirectionalSequenceRNN.cpp154 NN_RET_CHECK(!context->isOmittedOutput(kStateOutputTensor)); in prepare()
/packages/modules/NeuralNetworks/common/include/
DOperationsExecutionUtils.h58 virtual bool isOmittedOutput(uint32_t index) const = 0;
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp89 bool isOmittedOutput(uint32_t index) const override;
252 bool OperationExecutionContext::isOmittedOutput(uint32_t index) const { in isOmittedOutput() function in android::nn::__anonfe513d680111::OperationExecutionContext
262 NN_RET_CHECK(!isOmittedOutput(i)) in checkNoOmittedOperand()