Skip to content

Commit f9979c7

Browse files
committed
Update name
1 parent e307282 commit f9979c7

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tensorlayerx/nn/core/core_tensorflow.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -583,7 +583,7 @@ def init_build(self, *inputs, **kwargs):
583583
def str_to_init(self, initializer):
584584
return str2init(initializer)
585585

586-
def node_build(self, *inputs, **kwargs):
586+
def build_graph(self, *inputs, **kwargs):
587587
# Add nodes only when the composition is needed.
588588
layers = self.layers_and_names(name_prefix='')
589589
for layer_name, layer in layers:
@@ -680,10 +680,8 @@ def _construct_graph(self):
680680
indegrees[out_node.node_name] -= 1
681681
if indegrees[out_node.node_name] == 0:
682682
next_depth.append(out_node)
683-
684683
cur_depth = next_depth
685684
next_depth = []
686-
687685
return node_by_depth, all_layers
688686

689687

0 commit comments

Comments
 (0)