We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 38f5a48 commit 80bc0d8Copy full SHA for 80bc0d8
tf2onnx/graph.py
@@ -111,7 +111,9 @@ def get_value_attr(self, external_tensor_storage=None):
111
return a
112
if np.product(a.t.dims) > external_tensor_storage.external_tensor_size_threshold:
113
a = copy.copy(a)
114
- tensor_name = self.name + "_" + str(external_tensor_storage.name_counter)
+ tensor_name = self.name.strip() + "_" + str(external_tensor_storage.name_counter)
115
+ for c in '~"#%&*:<>?/\\{|}':
116
+ tensor_name = tensor_name.replace(c, '_')
117
external_tensor_storage.name_counter += 1
118
external_tensor_storage.name_to_tensor_data[tensor_name] = a.t.raw_data
119
external_tensor_storage.node_to_modified_value_attr[self] = a
0 commit comments