Home
last modified time | relevance | path

Searched refs:pad_after (Results 1 – 25 of 291) sorted by relevance

12345678910>>...12

/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/nn/
H A Dpad.py24 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 Dconv2d.py219 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 Dpad.py24 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 Dconv2d.py219 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 Dpad.py26 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 Dconv2d.py242 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 Dconv1d.py109 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 Dcorrelation.py79 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 Dconv3d.py83 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 Dbitserial_conv2d.py89 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 Darraypad.py1400 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 Dtest_pass_inject_copy_intrin.py31 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 Dtest_tir_transform_inject_copy_intrin.py35 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 Dtest_pass_inject_copy_intrin.py31 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 Dlocal_response_norm.h63 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 Dpooling.h99 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 Dlocal_response_norm.h63 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 Dpooling.h99 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 Ddilation2d.py86 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 Dlocal_response_norm.h60 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 Dpooling.h97 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 Dpad.cc89 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 Dinject_copy_intrin.cc111 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 Dinject_copy_intrin.cc111 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 Dinject_copy_intrin.cc108 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()

12345678910>>...12