NONATOMIC,
'l1_normalization':
NONATOMIC,
'l2_normalization':
NONATOMIC,
'batch_normalization':
convert_batch_normalization,
'avg_roi_pool':
UNSUPPORTED,
'max_roi_pool':
UNSUPPORTED, # maybe supported
'roi_resample':
UNSUPPORTED,
'avg_roi_align':
UNSUPPORTED,
'max_roi_align':
UNSUPPORTED,
'linear_quantize':
NONATOMIC,
'logarithmic_quantize':
NONATOMIC,
'copy_n':
convert_copy_n,
}
# TODO add to class as static(?) method
# NNEF must be parsed with this before calling nnef_to_tf.Converter on it
ParserConfig = NNEFParserConfig(lowered=[
k for k, v in six.iteritems(_DefaultConverters) if v is NONATOMIC
])
'select': convert_select,
'conv': convert_conv,
'deconv': convert_deconv,
'split': convert_split,
'linear': convert_linear,
'reshape': convert_reshape,
'leaky_relu': convert_leaky_relu,
'prelu': convert_prelu,
'local_response_normalization': convert_local_response_normalization,
'matmul': convert_matmul,
'transpose': convert_transpose,
'add_n': convert_add_n,
'softmax': convert_softmax,
'batch_normalization': convert_batch_normalization,
'squeeze': convert_squeeze,
'unsqueeze': convert_unsqueeze,
'max_pool_with_index': convert_max_pool_with_index,
'clamp': convert_clamp,
'slice': convert_slice,
'nearest_upsample': convert_nearest_upsample,
'nearest_downsample': convert_nearest_downsample,
"sin": partial(generic_convert_unary, target_name='Sin'),
"cos": partial(generic_convert_unary, target_name='Cos'),
"tile": convert_tile,
"pad": convert_pad,
}
# NNEF must be parsed with this before calling nnef_to_caffe.Converter on it
ParserConfig = NNEFParserConfig(lowered=[k for k, v in six.iteritems(_StandardConverters) if v is NONATOMIC])
