experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Set name of layer to node name when converting elementwise ops.

Browse Source 
PiperOrigin-RevId: 295242050
Change-Id: Id46bccd71a64770570e21fd2912e0c0c73ca8651
This commit is contained in:
A. Unique TensorFlower 2020-02-14 15:37:23 -08:00 committed by TensorFlower Gardener
parent 767e4d5dab
commit ad4ca90bc0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
View File

@ -4093,6 +4093,7 @@ Status ConvertBinary(OpConverterParams* params) {
nvinfer1::ILayer* layer = params->converter->network()->addElementWise(
*tensor_l, *tensor_r, op_pair->second);
TFTRT_RETURN_ERROR_IF_NULLPTR(layer, node_def.name());
layer->setName(node_def.name().c_str());
nvinfer1::ITensor* trt_tensor = layer->getOutput(0);
#if IS_TRT_VERSION_GE(5, 1, 0, 0)