Searched refs:tfLiteShapeInputTensor (Results 1 – 1 of 1) sorted by relevance
168 const TfLiteTensor& tfLiteShapeInputTensor = tfLiteTensors[tfLiteNode->inputs->data[1]]; in VisitReshapeOperator() local169 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()