Searched refs:m_TensorsInfo (Results 1 – 2 of 2) sorted by relevance
485 m_TensorsInfo[input].m_dtype, in ValidateInputs()508 return (m_TensorsInfo.count(name) == 0 || in ComputeOutputInfo()509 m_TensorsInfo[name].m_info == nullptr || in ComputeOutputInfo()510 … m_TensorsInfo[name].m_info->GetShape().GetDimensionality() == in ComputeOutputInfo()544 m_TensorsInfo[outNames[i]] = OnnxTensor(); in ComputeOutputInfo()545 m_TensorsInfo[outNames[i]].m_info = std::make_unique<TensorInfo>( in ComputeOutputInfo()547 m_TensorsInfo[outNames[i]].m_dtype = dataType; in ComputeOutputInfo()549 outInfo.push_back(*m_TensorsInfo[outNames[i]].m_info); in ComputeOutputInfo()570 m_TensorsInfo.clear(); in Cleanup()604 TensorInfo tensorInfo = *m_TensorsInfo[name].m_info; in CreateConstTensor()[all …]
193 std::unordered_map<std::string, OnnxTensor> m_TensorsInfo; member in armnnOnnxParser::OnnxParserImpl