Home
last modified time | relevance | path

Searched refs:weight_node_name (Results 1 – 2 of 2) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/tensorrt/
H A Dnnvm_to_onnx.cc488 std::string weight_node_name = ig[inputs[op::conv::kWeight].node_id].source->attrs.name; in ConvertFullyConnected() local
491 tranpose_node_proto->add_input(weight_node_name); in ConvertFullyConnected()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/tensorrt/
H A Dnnvm_to_onnx.cc488 std::string weight_node_name = ig[inputs[op::conv::kWeight].node_id].source->attrs.name; in ConvertFullyConnected() local
491 tranpose_node_proto->add_input(weight_node_name); in ConvertFullyConnected()