Searched refs:matmulNode (Results 1 – 2 of 2) sorted by relevance
120 …void AddFullyConnected(const onnx::NodeProto& matmulNode, const onnx::NodeProto* addNode = nullptr…
1150 void OnnxParserImpl::AddFullyConnected(const onnx::NodeProto& matmulNode, const onnx::NodeProto* ad… in AddFullyConnected() argument1157 CHECK_VALID_SIZE(static_cast<size_t>(matmulNode.input_size()), 2); in AddFullyConnected()1158 CHECK_VALID_SIZE(static_cast<size_t>(matmulNode.output_size()), 1); in AddFullyConnected()1159 VALID_INPUTS(matmulNode, STR_LIST(onnx::TensorProto::FLOAT)); in AddFullyConnected()1161 GetInputAndParam(matmulNode, &inputName, &weightName, CHECK_LOCATION()); in AddFullyConnected()1206 outputName = matmulNode.output(0); in AddFullyConnected()1210 layer = m_Network->AddFullyConnectedLayer(desc, matmulNode.name().c_str()); in AddFullyConnected()