提交 bc074d0e 编写于 作者: D dangqingqing

minor change

上级 2c2347df
4 合并请求!11636[IMPORTANT] MKLDNN layout: Support for sum operator,!2081Release/0.10.0,!1463Feature/reordered packages,!1374Data Feeder
...@@ -58,13 +58,13 @@ class DataFeeder(DataProviderConverter): ...@@ -58,13 +58,13 @@ class DataFeeder(DataProviderConverter):
:type reader_dict: dict() :type reader_dict: dict()
""" """
self.input_names = [] self.input_names = []
self.input_types = [] input_types = []
self.reader_dict = reader_dict self.reader_dict = reader_dict
for each in data_types: for each in data_types:
self.input_names.append(each[0]) self.input_names.append(each[0])
assert isinstance(each[1], data_type.InputType) assert isinstance(each[1], data_type.InputType)
self.input_types.append(each[1]) input_types.append(each[1])
DataProviderConverter.__init__(self, self.input_types) DataProviderConverter.__init__(self, input_types)
def convert(self, dat, argument=None): def convert(self, dat, argument=None):
""" """
...@@ -98,6 +98,3 @@ class DataFeeder(DataProviderConverter): ...@@ -98,6 +98,3 @@ class DataFeeder(DataProviderConverter):
return retv return retv
return DataProviderConverter.convert(self, reorder_data(dat), argument) return DataProviderConverter.convert(self, reorder_data(dat), argument)
def __call__(self, dat, argument=None):
return self.convert(dat, argument)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册
反馈
建议
客服 返回
顶部