def create_pooling_attrs(node, pool_method):
data_format = node.pb.attr["data_format"]
attrs = {
'auto_pad': convert_tf_padding_to_str(node.pb.attr['padding'].s.decode()),
'window': tf_int_list(node.pb.attr['ksize'].list),
'spatial_dims': tf_data_format_spatial(data_format),
'pad': None, # will be inferred when input shape is known
'stride': tf_int_list(node.pb.attr['strides'].list),
'pad_spatial_shape': None,
'output_spatial_shape': None,
'pool_method': pool_method,
'layout': data_format.s.decode(),
'exclude_pad': True,
}
return attrs
def tf_create_attrs(node, input_feature_channel, output_feature_channel):
data_format = node.pb.attr["data_format"]
return {
'auto_pad': convert_tf_padding_to_str(node.pb.attr['padding']),
'bias_addable': True,
'bias_term': False,
'spatial_dims': tf_data_format_spatial(data_format),
'channel_dims': tf_data_format_channel(data_format),
'batch_dims': tf_data_format_batch(data_format),
'pad': None, # will be inferred when input shape is known
'pad_spatial_shape': None,
'output_spatial_shape': None,
'output_shape': None,
'output': None,
'stride': tf_int_list(node.pb.attr["strides"].list),
'type': None, # don't set type until we are sure it is really translated to correct IR; see infer function
'group': None,
'layout': data_format.s.decode(),
'input_feature_channel': input_feature_channel,
'output_feature_channel': output_feature_channel,
}
