Searched refs:axisTensorData (Results 1 – 2 of 2) sorted by relevance
251 auto* axisTensorData = tflite::GetTensorData<int32_t>(&tfLiteAxisTensor); in VisitMeanOperator() local257 axis.emplace_back(axisTensorData[i]); in VisitMeanOperator()
2233 std::vector<int32_t> axisTensorData; in ParseConcat() local2234 shapeNode->GetConstTensor(axisTensorData); in ParseConcat()2237 const unsigned int concatDim = static_cast<unsigned int>(axisTensorData[0]); in ParseConcat()2780 std::vector<int32_t> axisTensorData; in ParseSplit() local2781 shapeNode->GetConstTensor(axisTensorData); in ParseSplit()2784 const unsigned int splitDim = static_cast<unsigned int>(axisTensorData[0]); in ParseSplit()