Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DRedefine.hpp168 const TfLiteTensor& tfLiteShapeInputTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitReshapeOperator() local
169 if (!IsValid(tfLiteContext, tfLiteShapeInputTensor, operatorCode, nodeIndex)) in VisitReshapeOperator()
174 if (tfLiteShapeInputTensor.dims->size != 1) in VisitReshapeOperator()
184 auto* shapeTensorDataPtr = tflite::GetTensorData<int32_t>(&tfLiteShapeInputTensor); in VisitReshapeOperator()
185 auto shapeTensorNumValues = tfLiteShapeInputTensor.dims->data[0]; in VisitReshapeOperator()