Searched refs:SetLayerName (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 475 SetLayerName(layer, "_tftrt_constant_", in CreateConstantLayer()
1390 SetLayerName(layer, tensor->getName(), "shuffle", output_index); in RenameAndMarkOutputTensors()
1599 SetLayerName(layer, node_def, sub_op_name); in TransposeTensor()
1671 void Converter::SetLayerName(nvinfer1::ILayer* layer, const NodeDef& node_def, in SetLayerName() function in tensorflow::tensorrt::convert::Converter
1686 void Converter::SetLayerName(nvinfer1::ILayer* layer, in SetLayerName() function in tensorflow::tensorrt::convert::Converter
1740 converter->SetLayerName(layer, node_def, "shuffle", op_instance); in PrepareTensorForShape()
2243 params->converter->SetLayerName(pad_layer, params->node_def, "pad"); in Conv2DPaddingHelper()
2430 params->converter->SetLayerName(conv_layer, node_def, "conv"); in ConvertConv2DHelper()
2463 params->converter->SetLayerName(padding_layer, node_def, "pad"); in ConvertConv2DHelper()
2558 params->converter->SetLayerName(shape_layer, params->node_def, "shape"); in ConvertShape()
[all …]
|
| D | convert_nodes.h | 602 void SetLayerName(nvinfer1::ILayer* layer, const NodeDef& node_def,
605 void SetLayerName(nvinfer1::ILayer* layer, absl::string_view main_op_name,
|