def test_eltwise_mul(self): attrs = {} res = eltwise_ext(AttrDictionary(attrs), infer=lambda a, b: a * b, op_type="mul") exp_attrs = {'type': 'Eltwise', 'operation': 'mul'} for key in exp_attrs.keys(): self.assertEqual(res[key], exp_attrs[key])
mxnet_op_extractors = { 'BatchNorm': extractor_wrapper(batch_norm_ext), 'Crop': extractor_wrapper(crop_ext), 'ScaleShift': extractor_wrapper(scale_shift_ext), 'slice_axis': extractor_wrapper(slice_axis_ext), 'null': lambda node: null_ext(node.symbol_dict), 'Concat': extractor_wrapper(concat_ext), 'elemwise_add': extractor_wrapper(lambda attrs: eltwise_ext( attrs, infer=lambda a, b: a + b, op_type="sum")), 'elemwise_mul': extractor_wrapper(lambda attrs: eltwise_ext( attrs, infer=lambda a, b: a * b, op_type="mul")), '_Plus': extractor_wrapper(lambda attrs: eltwise_ext( attrs, infer=lambda a, b: a + b, op_type="sum")), 'FullyConnected': extractor_wrapper(fully_connected_ext), 'transpose': extractor_wrapper(transpose_ext), 'LRN': extractor_wrapper(lrn_ext), 'L2Normalization': extractor_wrapper(l2_normalization_ext), '_contrib_MultiBoxPrior':