Searched refs:shape_layer (Results 1 – 1 of 1) sorted by relevance
2555 nvinfer1::IShapeLayer* shape_layer = in ConvertShape() local2557 TFTRT_RETURN_ERROR_IF_NULLPTR(shape_layer, params->node_def.name()); in ConvertShape()2558 params->converter->SetLayerName(shape_layer, params->node_def, "shape"); in ConvertShape()2559 params->outputs->push_back(TRT_TensorOrWeights(shape_layer->getOutput(0))); in ConvertShape()