Home
last modified time | relevance | path

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

/external/armnn/delegate/classic/src/
DConvolution.hpp664 const TfLiteTensor& tfLiteOutputShapeTensor = tfLiteTensors[tfLiteNode->inputs->data[0]]; in VisitTransposeConv2dOperator() local
665 if(!IsValid(&tfLiteOutputShapeTensor)) in VisitTransposeConv2dOperator()
673 if (IsDynamicTensor(tfLiteOutputShapeTensor)) in VisitTransposeConv2dOperator()
682 …st armnn::TensorInfo outputShapeTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteOutputShapeTensor); in VisitTransposeConv2dOperator()
688 outputShape[i] = ::tflite::GetTensorData<int32_t>(&tfLiteOutputShapeTensor)[i]; in VisitTransposeConv2dOperator()
696 outputShape[i] = ::tflite::GetTensorData<uint8_t>(&tfLiteOutputShapeTensor)[i]; in VisitTransposeConv2dOperator()