MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / set_tensor_name

Function set_tensor_name

demo/BERT/builder_varseqlen.py:89–90  ·  view source on GitHub ↗
(tensor, prefix, name)

Source from the content-addressed store, hash-verified

87 return dtype
88
89def set_tensor_name(tensor, prefix, name):
90 tensor.name = prefix + name
91
92def set_output_name(layer, prefix, name, out_idx = 0):
93 set_tensor_name(layer.get_output(out_idx), prefix, name)

Callers 2

set_output_nameFunction · 0.70
transformer_layer_optFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected