Home
last modified time | relevance | path

Searched refs:equal_const_int (Results 1 – 14 of 14) sorted by relevance

/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/nn/
H A Dpad.py20 from ..util import equal_const_int
66 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0):
128 if equal_const_int(pad_before[i], 0) and equal_const_int(
H A Ddilate.py58 if not util.equal_const_int(strides[i], 1):
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/nn/
H A Dpad.py20 from ..util import equal_const_int
66 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0):
128 if equal_const_int(pad_before[i], 0) and equal_const_int(
H A Ddilate.py58 if not util.equal_const_int(strides[i], 1):
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/nn/
H A Dpad.py21 from ..util import equal_const_int
77 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0):
135 if equal_const_int(pad_before[i], 0) and equal_const_int(pad_after[i], 0):
H A Ddilate.py60 if not util.equal_const_int(strides[i], 1):
/dports/misc/tvm/incubator-tvm-0.6.1/vta/python/vta/
H A Dir_pass.py345 if not util.equal_const_int(size - buf.strides[i], 0):
358 if util.equal_const_int(x_size - elem_block, 0):
375 if not util.equal_const_int(idxm(x_stride, elem_block), 0):
417 flat = util.equal_const_int(shape[-1], elem_block)
420 if not util.equal_const_int(strides[-1], 1):
428 if not util.equal_const_int(strides[-2] - elem_block, 0):
446 if not util.equal_const_int(strides[-1], 1):
448 if not util.equal_const_int(strides[-2] - shape[-1], 0):
458 if not util.equal_const_int(strides[-3], elem_block):
541 if not util.equal_const_int(pad_after[dim], 0):
[all …]
/dports/misc/py-tvm/incubator-tvm-0.6.1/vta/python/vta/
H A Dir_pass.py345 if not util.equal_const_int(size - buf.strides[i], 0):
358 if util.equal_const_int(x_size - elem_block, 0):
375 if not util.equal_const_int(idxm(x_stride, elem_block), 0):
417 flat = util.equal_const_int(shape[-1], elem_block)
420 if not util.equal_const_int(strides[-1], 1):
428 if not util.equal_const_int(strides[-2] - elem_block, 0):
446 if not util.equal_const_int(strides[-1], 1):
448 if not util.equal_const_int(strides[-2] - shape[-1], 0):
458 if not util.equal_const_int(strides[-3], elem_block):
541 if not util.equal_const_int(pad_after[dim], 0):
[all …]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/vta/python/vta/
H A Dtransform.py349 if not util.equal_const_int(size - buf.strides[i], 0):
363 if util.equal_const_int(x_size - elem_block, 0):
382 if not util.equal_const_int(idxm(x_stride, elem_block), 0):
428 flat = util.equal_const_int(shape[-1], elem_block)
431 if not util.equal_const_int(strides[-1], 1):
439 if not util.equal_const_int(strides[-2] - elem_block, 0):
457 if not util.equal_const_int(strides[-1], 1):
459 if not util.equal_const_int(strides[-2] - shape[-1], 0):
469 if not util.equal_const_int(strides[-3], elem_block):
561 if not util.equal_const_int(pad_after[dim], 0):
[all …]
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/
H A Dconv2d_transpose_nchw.py24 from ..util import equal_const_int, get_const_tuple, traverse_inline
87 if not equal_const_int(strides[i], 1):
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/cuda/
H A Dconv2d_transpose_nchw.py24 from ..util import equal_const_int, get_const_tuple, traverse_inline
87 if not equal_const_int(strides[i], 1):
/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/
H A Dutil.py124 def equal_const_int(expr, value): function
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/
H A Dutil.py124 def equal_const_int(expr, value): function
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/
H A Dutil.py138 def equal_const_int(expr, value): function