/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/nn/ |
H A D | pad.py | 24 def pad(data, pad_before, pad_after=None, pad_value=0.0, name="PadInput"): argument 50 pad_after = pad_after if pad_after else pad_before 54 if len(pad_after) != n: 59 (data.shape[i] + pad_before[i] + pad_after[i])) for i in range(n)) 66 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0): 82 pad_after=None, argument 110 pad_after = pad_after if pad_after else pad_before 114 if len(pad_after) != n: 118 tvm.ir_pass.Simplify((data.shape[i] + pad_before[i] + pad_after[i])) 129 pad_after[i], 0):
|
H A D | conv2d.py | 219 pad_after = [0, 0, pad_down, pad_right] 220 temp = pad(Input, pad_before, pad_after, name="pad_temp") 285 pad_after = [pad_down, pad_right, 0, 0] 286 PaddedInput = pad(Input, pad_before, pad_after, name="PaddedInput") 349 pad_after = [0, pad_down, pad_right, 0] 350 PaddedInput = pad(Input, pad_before, pad_after, name="PaddedInput") 830 pad_after = [0, 0, pad_down, pad_right] 831 temp = pad(Input, pad_before, pad_after, name="pad_temp")
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/nn/ |
H A D | pad.py | 24 def pad(data, pad_before, pad_after=None, pad_value=0.0, name="PadInput"): argument 50 pad_after = pad_after if pad_after else pad_before 54 if len(pad_after) != n: 59 (data.shape[i] + pad_before[i] + pad_after[i])) for i in range(n)) 66 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0): 82 pad_after=None, argument 110 pad_after = pad_after if pad_after else pad_before 114 if len(pad_after) != n: 118 tvm.ir_pass.Simplify((data.shape[i] + pad_before[i] + pad_after[i])) 129 pad_after[i], 0):
|
H A D | conv2d.py | 219 pad_after = [0, 0, pad_down, pad_right] 220 temp = pad(Input, pad_before, pad_after, name="pad_temp") 285 pad_after = [pad_down, pad_right, 0, 0] 286 PaddedInput = pad(Input, pad_before, pad_after, name="PaddedInput") 349 pad_after = [0, pad_down, pad_right, 0] 350 PaddedInput = pad(Input, pad_before, pad_after, name="PaddedInput") 830 pad_after = [0, 0, pad_down, pad_right] 831 temp = pad(Input, pad_before, pad_after, name="pad_temp")
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/nn/ |
H A D | pad.py | 26 def pad(data, pad_before, pad_after=None, pad_value=0.0, name="PadInput"): argument 52 pad_after = pad_after if pad_after else pad_before 57 if len(pad_after) != n: 66 out_shape = tuple(ana.simplify(dshape[i] + pad_before[i] + pad_after[i]) for i in range(n)) 77 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0): 92 def mirror_pad(data, pad_before, pad_after=None, mode="SYMMETRIC", name="MirrorPadInput"): argument 118 pad_after = pad_after if pad_after else pad_before 123 if len(pad_after) != n: 126 out_shape = tuple(ana.simplify(data.shape[i] + pad_before[i] + pad_after[i]) for i in range(n)) 135 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0):
|
H A D | conv2d.py | 242 pad_after = [0, 0, pad_down, pad_right] 243 temp = pad(Input, pad_before, pad_after, name="pad_temp") 314 pad_after = [pad_down, pad_right, 0, 0] 386 pad_after = [0, pad_down, pad_right, 0] 473 pad_after = (0, 0, pad_down, pad_right, 0) 478 data_pad = pad(data, pad_before, pad_after, name="data_pad") 577 pad_after = (0, 0, pad_down, pad_right, 0) 809 pad_after = [0, 0, pad_down, pad_right] 810 temp = pad(Input, pad_before, pad_after, name="pad_temp") 894 pad_after = [0, pad_down, pad_right, 0] [all …]
|
H A D | conv1d.py | 109 pad_after = [0, 0, pad_right] 110 temp = pad(data, pad_before, pad_after, name="pad_temp") 169 pad_after = [0, pad_right, 0] 170 temp = pad(data, pad_before, pad_after, name="pad_temp")
|
H A D | correlation.py | 79 pad_after = [0, 0, pad_after_h, pad_after_w] 80 padded_data1 = pad(data1, pad_before, pad_after) 81 padded_data2 = pad(data2, pad_before, pad_after)
|
H A D | conv3d.py | 83 pad_after = [0, 0, pad_back, pad_down, pad_right] 84 temp = pad(Input, pad_before, pad_after, name="pad_temp") 160 pad_after = [0, pad_back, pad_down, pad_right, 0] 161 PaddedInput = pad(Input, pad_before, pad_after, name="PaddedInput")
|
H A D | bitserial_conv2d.py | 89 pad_after = [0, 0, 0, DPAD, RPAD] 91 PadInput_q = pad(Input_q, pad_before, pad_after, name="pad_temp") 213 pad_after = [0, DPAD, RPAD, 0, 0] 223 PadInput_q = pad(Input_q, pad_before, pad_after, name="PaddedInput")
|
/dports/science/py-scikit-fuzzy/scikit-fuzzy-0.4.2/skfuzzy/image/ |
H A D | arraypad.py | 1400 for axis, (pad_before, pad_after) in enumerate(pad_width): 1402 newmat = _append_edge(newmat, pad_after, axis) 1435 for axis, (pad_before, pad_after) in enumerate(pad_width): 1440 (pad_after > 0)) and newmat.shape[axis] == 1: 1444 newmat = _append_edge(newmat, pad_after, axis) 1456 pad_after -= pad_iter_a 1461 for axis, (pad_before, pad_after) in enumerate(pad_width): 1468 (pad_after > safe_pad)): 1475 pad_after -= pad_iter_a 1486 (pad_after > safe_pad)): [all …]
|
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/unittest/ |
H A D | test_pass_inject_copy_intrin.py | 31 def cb(src, dst, pad_before, pad_after, pad_value): argument 53 def cb(src, dst, pad_before, pad_after, pad_value): argument 57 assert pad_after[0].value == 1 58 assert pad_after[1].value == 0 74 def cb(src, dst, pad_before, pad_after, pad_value): argument 103 def cb(src, dst, pad_before, pad_after, pad_value): argument 110 assert_expr_equal(pad_after[0], rpad_after)
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/unittest/ |
H A D | test_tir_transform_inject_copy_intrin.py | 35 def cb(src, dst, pad_before, pad_after, pad_value): argument 63 def cb(src, dst, pad_before, pad_after, pad_value): argument 67 assert pad_after[0].value == 1 68 assert pad_after[1].value == 0 87 def cb(src, dst, pad_before, pad_after, pad_value): argument 116 def cb(src, dst, pad_before, pad_after, pad_value): argument 123 tvm.testing.assert_prim_expr_equal(pad_after[0], rpad_after)
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/unittest/ |
H A D | test_pass_inject_copy_intrin.py | 31 def cb(src, dst, pad_before, pad_after, pad_value): argument 53 def cb(src, dst, pad_before, pad_after, pad_value): argument 57 assert pad_after[0].value == 1 58 assert pad_after[1].value == 0 74 def cb(src, dst, pad_before, pad_after, pad_value): argument 103 def cb(src, dst, pad_before, pad_after, pad_value): argument 110 assert_expr_equal(pad_after[0], rpad_after)
|
/dports/misc/tvm/incubator-tvm-0.6.1/topi/include/topi/nn/ |
H A D | local_response_norm.h | 63 Array<Expr> pad_after{ 0, 0, 0, 0}; 65 pad_after.Set(axis, static_cast<Expr>(size/2)); 66 auto pad_data = pad(data, pad_before, pad_after, 0, "pad_data");
|
H A D | pooling.h | 99 Array<Expr> pad_after(std::vector<Expr>(x->shape.size(), 0)); in pool_impl() 100 pad_after.Set(height_axis, pad_bottom); in pool_impl() 101 pad_after.Set(width_axis, pad_right); in pool_impl() 123 auto temp = do_pad ? pad(x, pad_before, pad_after, x->dtype.min(), "pad_temp") : x; in pool_impl() 133 auto temp = do_pad ? pad(x, pad_before, pad_after, 0, "pad_temp") : x; in pool_impl() 205 Array<Expr> pad_after(std::vector<Expr>(x->shape.size(), 0)); in pool_grad_impl() 206 pad_after.Set(height_axis, pad_bottom); in pool_grad_impl() 207 pad_after.Set(width_axis, pad_right); in pool_grad_impl() 237 auto pad_x = do_pad ? pad(x, pad_before, pad_after, x->dtype.min(), "pad_temp") : x; in pool_grad_impl()
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/include/topi/nn/ |
H A D | local_response_norm.h | 63 Array<Expr> pad_after{ 0, 0, 0, 0}; 65 pad_after.Set(axis, static_cast<Expr>(size/2)); 66 auto pad_data = pad(data, pad_before, pad_after, 0, "pad_data");
|
H A D | pooling.h | 99 Array<Expr> pad_after(std::vector<Expr>(x->shape.size(), 0)); in pool_impl() 100 pad_after.Set(height_axis, pad_bottom); in pool_impl() 101 pad_after.Set(width_axis, pad_right); in pool_impl() 123 auto temp = do_pad ? pad(x, pad_before, pad_after, x->dtype.min(), "pad_temp") : x; in pool_impl() 133 auto temp = do_pad ? pad(x, pad_before, pad_after, 0, "pad_temp") : x; in pool_impl() 205 Array<Expr> pad_after(std::vector<Expr>(x->shape.size(), 0)); in pool_grad_impl() 206 pad_after.Set(height_axis, pad_bottom); in pool_grad_impl() 207 pad_after.Set(width_axis, pad_right); in pool_grad_impl() 237 auto pad_x = do_pad ? pad(x, pad_before, pad_after, x->dtype.min(), "pad_temp") : x; in pool_grad_impl()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/image/ |
H A D | dilation2d.py | 86 pad_after = [0, 0, pad_down, pad_right] 87 temp = pad(input, pad_before, pad_after, name="pad_temp") 162 pad_after = [0, pad_down, pad_right, 0] 163 padded_input = pad(input, pad_before, pad_after, name="padded_input")
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/topi/nn/ |
H A D | local_response_norm.h | 60 Array<PrimExpr> pad_after{0, 0, 0, 0}; 62 pad_after.Set(axis, static_cast<PrimExpr>(size / 2)); 63 auto pad_data = pad(data, pad_before, pad_after, 0, "pad_data");
|
H A D | pooling.h | 97 Array<PrimExpr> pad_after(std::vector<PrimExpr>(x->shape.size(), 0)); in pool_impl() 98 pad_after.Set(height_axis, pad_bottom); in pool_impl() 99 pad_after.Set(width_axis, pad_right); in pool_impl() 121 auto temp = do_pad ? pad(x, pad_before, pad_after, tvm::min_value(x->dtype), "pad_temp") : x; in pool_impl() 134 auto temp = do_pad ? pad(x, pad_before, pad_after, 0, "pad_temp") : x; in pool_impl() 211 Array<PrimExpr> pad_after(std::vector<PrimExpr>(x->shape.size(), 0)); in pool_grad_impl() 212 pad_after.Set(height_axis, pad_bottom); in pool_grad_impl() 213 pad_after.Set(width_axis, pad_right); in pool_grad_impl() 663 Array<PrimExpr> pad_after(std::vector<PrimExpr>(x_size, 0)); in pool_impl_nd() 686 pad_after.Set(ii, pad_tail[i]); in pool_impl_nd() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/relay/op/dyn/nn/ |
H A D | pad.cc | 89 Array<IndexExpr> pad_after; in PadCompute() local 93 pad_after.push_back(pad_width[i][1]); in PadCompute() 99 return Array<te::Tensor>{topi::pad(inputs[0], pad_before, pad_after, pad_value, "T_pad", in PadCompute()
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/pass/ |
H A D | inject_copy_intrin.cc | 111 Array<Expr> pad_before, pad_after; in MatchCopyPattern() local 138 pad_after.push_back(pafter); in MatchCopyPattern() 140 pad_after.push_back(make_zero(t)); in MatchCopyPattern() 172 *out = flower_copy_fromto_(src, dst, pad_before, pad_after, pad_value); in MatchCopyPattern()
|
/dports/misc/tvm/incubator-tvm-0.6.1/src/pass/ |
H A D | inject_copy_intrin.cc | 111 Array<Expr> pad_before, pad_after; in MatchCopyPattern() local 138 pad_after.push_back(pafter); in MatchCopyPattern() 140 pad_after.push_back(make_zero(t)); in MatchCopyPattern() 172 *out = flower_copy_fromto_(src, dst, pad_before, pad_after, pad_value); in MatchCopyPattern()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/tir/transforms/ |
H A D | inject_copy_intrin.cc | 108 Array<PrimExpr> pad_before, pad_after; in MatchCopyPattern() local 134 pad_after.push_back(pafter); in MatchCopyPattern() 136 pad_after.push_back(make_zero(t)); in MatchCopyPattern() 156 *out = flower_copy_fromto_(src, dst, pad_before, pad_after, pad_value); in MatchCopyPattern()
|