Searched defs:AddActivation (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/ | ||
| D | activation_tensorrt.cc | 73 nvinfer1::IActivationLayer *ActivationTensorRT::AddActivation(nvinfer1::INetworkDefinition *network, in AddActivation() function in mindspore::lite::ActivationTensorRT |
| D | elementwise_tensorrt.cc | 162 nvinfer1::ITensor *ElementWiseTensorRT::AddActivation(nvinfer1::INetworkDefinition *network, in AddActivation() function in mindspore::lite::ElementWiseTensorRT |