Function WillInputBeOptimizedToConst

delegate/classic/src/SharedFunctions.cpp:152–189

}

bool WillInputBeOptimizedToConst(TfLiteContext* tfLiteContext, int32_t inputIdx)
{
    int32_t connectedIndex;
    TfLiteNode* connectedNode = GetNodeConnectedToInput(tfLiteContext, connectedIndex, inputIdx);

    if (connectedNode)
    {
        TfLiteRegistration* tfLiteRegistration = nullptr;

        if (tfLiteContext->GetNodeAndRegistration(tfLiteContext, connectedIndex, &connectedNode, &tfLiteRegistration)
            == kTfLiteOk)
        {
            switch (tfLiteRegistration->builtin_code)
            {
                case kTfLiteBuiltinDequantize:
                {
                    if (connectedNode->inputs->size >= 1)
                    {
                        const TfLiteTensor* tfLiteTensors = tfLiteContext->tensors;
                        const TfLiteTensor& tfLiteInputTensor = tfLiteTensors[connectedNode->inputs->data[0]];

                        // If the input to the Dequantize is a Constant then both that Constant layer and the Dequantize
                        // layer will be replaced by a single Constant layer containing the dequantized values.
                        if (tflite::IsConstantTensor(&tfLiteInputTensor))
                        {
                            return true;
                        }
                    }
                    break;
                }
                default:
                {
                }
            }
        }
    }
    return false;
}

} // namespace armnnDelegate