diff --git a/python/paddle/fluid/async_executor.py b/python/paddle/fluid/async_executor.py index 7896a61f02ecbb20e0163343e2831d373025293f..671fe463e570211365af07d9c3c95dad895f61a3 100644 --- a/python/paddle/fluid/async_executor.py +++ b/python/paddle/fluid/async_executor.py @@ -45,6 +45,10 @@ class DataFeedDesc(object): def get_slot(self, name): return self.proto_desc.multi_slot_desc.slots[self.__name_to_index[name]] + def set_dense_slots(self, dense_slots_name): + for name in dense_slots_name: + self.proto_desc.multi_slot_desc.slots[self.__name_to_index[name]].dense = True + def set_use_slots(self, use_slots_name): for name in use_slots_name: self.proto_desc.multi_slot_desc.slots[self.__name_to_index[name]].use = True