File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
python/paddle/fluid/transpiler Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -410,7 +410,8 @@ def transpile(self,
410
410
attrs = {"axis" : 0 })
411
411
412
412
if self .has_distributed_lookup_table :
413
- self ._replace_lookup_table_op_with_prefetch (program , pserver_endpoints )
413
+ self ._replace_lookup_table_op_with_prefetch (program ,
414
+ pserver_endpoints )
414
415
self ._split_table_grad_and_add_send_vars (program , pserver_endpoints )
415
416
416
417
def get_trainer_program (self ):
@@ -631,7 +632,8 @@ def _get_splited_name_and_shape(varname):
631
632
return s_prog
632
633
633
634
# transpiler function for dis lookup_table
634
- def _replace_lookup_table_op_with_prefetch (self , program , pserver_endpoints ):
635
+ def _replace_lookup_table_op_with_prefetch (self , program ,
636
+ pserver_endpoints ):
635
637
# 1. replace lookup_table_op with split_ids_op -> prefetch_op -> sum_op
636
638
self .prefetch_input_vars = None
637
639
self .prefetch_output_vars = None
You can’t perform that action at this time.
0 commit comments