Home
last modified time | relevance | path

Searched refs:mightNeedPreprocessNode (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.cc97 auto mightNeedPreprocessNode = preprocess_map.find(source->op()->name); in ConvertNnvmGraphToOnnx() local
99 if (mightNeedPreprocessNode != preprocess_map.end()) { in ConvertNnvmGraphToOnnx()
100 mightNeedPreprocessNode->second(source->attrs, source->inputs, params_map); in ConvertNnvmGraphToOnnx()
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/subgraph/tensorrt/
H A Dnnvm_to_onnx.cc97 auto mightNeedPreprocessNode = preprocess_map.find(source->op()->name); in ConvertNnvmGraphToOnnx() local
99 if (mightNeedPreprocessNode != preprocess_map.end()) { in ConvertNnvmGraphToOnnx()
100 mightNeedPreprocessNode->second(source->attrs, source->inputs, params_map); in ConvertNnvmGraphToOnnx()