|
|
|
@ -117,7 +117,9 @@ class Layer(object):
|
|
|
|
|
continue
|
|
|
|
|
layer.to_proto(context=context)
|
|
|
|
|
|
|
|
|
|
if self.use_context_name():
|
|
|
|
|
if self.context_name() is None:
|
|
|
|
|
return ret_val
|
|
|
|
|
elif self.use_context_name():
|
|
|
|
|
return context[self.context_name()]
|
|
|
|
|
else:
|
|
|
|
|
return context[self.name]
|
|
|
|
|