Searched refs:m_InputShapes (Results 1 – 6 of 6) sorted by relevance
195 m_InputShapes(inputShapes),228 std::vector<armnn::TensorShape> m_InputShapes; member in __anonfedd02e00111::ArmnnConverter240 if (!m_InputShapes.empty()) in CreateNetwork()242 const size_t numInputShapes = m_InputShapes.size(); in CreateNetwork()253 inputShapes[m_InputNames[i]] = m_InputShapes[i]; in CreateNetwork()273 if (!m_InputShapes.empty()) in CreateNetwork()275 const size_t numInputShapes = m_InputShapes.size(); in CreateNetwork()300 if (!m_InputShapes.empty()) in CreateNetwork()302 const size_t numInputShapes = m_InputShapes.size(); in CreateNetwork()
65 std::vector<armnn::TensorShape> m_InputShapes; member134 if (!params.m_InputShapes.empty()) in Create()136 const size_t numInputShapes = params.m_InputShapes.size(); in Create()147 inputShapes[params.m_InputBindings[i]] = params.m_InputShapes[i]; in Create()299 if (!params.m_InputShapes.empty()) in Create()301 const size_t numInputShapes = params.m_InputShapes.size(); in Create()312 inputShapes[params.m_InputBindings[i]] = params.m_InputShapes[i]; in Create()
401 modelParams.m_InputShapes.push_back(*inputTensorShape);
61 modelParams.m_InputShapes = { inputTensorShape }; in main()
573 m_InputShapes.clear(); in Cleanup()741 m_InputShapes = inputShapes; in CreateNetworkFromTextFile()757 m_InputShapes = inputShapes; in CreateNetworkFromBinary()823 m_InputShapes = inputShapes; in CreateNetworkFromBinaryFile()858 m_InputShapes = inputShapes; in CreateNetworkFromString()2387 if (m_InputShapes.find(input.name()) == m_InputShapes.end()) in SetupInputLayers()2396 tensorInfo.SetShape(m_InputShapes[input.name()]); in SetupInputLayers()
227 std::map<std::string, armnn::TensorShape> m_InputShapes; member in armnnOnnxParser::OnnxParserImpl