/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/vta/python/vta/top/ |
H A D | graphpack.py | 51 int(dshape[2]), 73 dshape = tuple([dshape[0] + pad_width, dshape[1], dshape[2]]) 74 return data, dshape 85 dshape = tuple([dshape[0] + pad_width, dshape[1], dshape[2], dshape[3]]) 101 dshape = tuple(dshape[0], [dshape[1] + pad_width, dshape[2], dshape[3]]) 131 dshape = _to_shape(dshape) 142 dshape[2], 143 dshape[3], 152 dshape = _to_shape(dshape) 155 data = op.reshape(data, newshape=(dshape[0] // cfactor, cfactor, dshape[1], dshape[2], 1)) [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | bilinear_resize-inl.h | 284 new_height = ((dshape[2] % 2) == 0) ? dshape[2] : dshape[2] - 1; in BilinearSampleOpInferShape() 285 new_width = ((dshape[3] % 2) == 0) ? dshape[3] : dshape[3] - 1; in BilinearSampleOpInferShape() 290 new_height = ((dshape[2] % 2) == 0) ? dshape[2] : dshape[2] + 1; in BilinearSampleOpInferShape() 291 new_width = ((dshape[3] % 2) == 0) ? dshape[3] : dshape[3] + 1; in BilinearSampleOpInferShape() 296 new_height = ((dshape[2] % 2) == 1) ? dshape[2] : dshape[2] - 1; in BilinearSampleOpInferShape() 297 new_width = ((dshape[3] % 2) == 1) ? dshape[3] : dshape[3] - 1; in BilinearSampleOpInferShape() 302 new_height = ((dshape[2] % 2) == 1) ? dshape[2] : dshape[2] + 1; in BilinearSampleOpInferShape() 303 new_width = ((dshape[3] % 2) == 1) ? dshape[3] : dshape[3] + 1; in BilinearSampleOpInferShape() 312 dshape[2] = new_height; in BilinearSampleOpInferShape() 313 dshape[3] = new_width; in BilinearSampleOpInferShape() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
H A D | bilinear_resize-inl.h | 284 new_height = ((dshape[2] % 2) == 0) ? dshape[2] : dshape[2] - 1; in BilinearSampleOpInferShape() 285 new_width = ((dshape[3] % 2) == 0) ? dshape[3] : dshape[3] - 1; in BilinearSampleOpInferShape() 290 new_height = ((dshape[2] % 2) == 0) ? dshape[2] : dshape[2] + 1; in BilinearSampleOpInferShape() 291 new_width = ((dshape[3] % 2) == 0) ? dshape[3] : dshape[3] + 1; in BilinearSampleOpInferShape() 296 new_height = ((dshape[2] % 2) == 1) ? dshape[2] : dshape[2] - 1; in BilinearSampleOpInferShape() 297 new_width = ((dshape[3] % 2) == 1) ? dshape[3] : dshape[3] - 1; in BilinearSampleOpInferShape() 302 new_height = ((dshape[2] % 2) == 1) ? dshape[2] : dshape[2] + 1; in BilinearSampleOpInferShape() 303 new_width = ((dshape[3] % 2) == 1) ? dshape[3] : dshape[3] + 1; in BilinearSampleOpInferShape() 312 dshape[2] = new_height; in BilinearSampleOpInferShape() 313 dshape[3] = new_width; in BilinearSampleOpInferShape() [all …]
|
/dports/misc/tvm/incubator-tvm-0.6.1/vta/python/vta/top/ |
H A D | nnvm_graphpack.py | 26 assert dshape[0] % bfactor == 0 27 assert dshape[1] % cfactor == 0 31 dshape[2], dshape[3])) 48 assert len(dshape) == 4 49 assert dshape[0] % cfactor == 0 50 assert dshape[1] % cfactor == 0 54 dshape[2], dshape[3])) 63 assert len(dshape) == 4 69 dshape[2], dshape[3])) 78 assert len(dshape) == 3 [all …]
|
H A D | graphpack.py | 43 int(dshape[2]), int(dshape[3]))) 60 assert len(dshape) == 4 66 int(dshape[2]), int(dshape[3]))) 75 dshape = _to_shape(dshape) 76 assert len(dshape) == 4 77 assert dshape[0] % cfactor == 0 78 assert dshape[1] % cfactor == 0 82 dshape[2], dshape[3])) 91 dshape = _to_shape(dshape) 92 assert len(dshape) == 3 [all …]
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/vta/python/vta/top/ |
H A D | nnvm_graphpack.py | 26 assert dshape[0] % bfactor == 0 27 assert dshape[1] % cfactor == 0 31 dshape[2], dshape[3])) 48 assert len(dshape) == 4 49 assert dshape[0] % cfactor == 0 50 assert dshape[1] % cfactor == 0 54 dshape[2], dshape[3])) 63 assert len(dshape) == 4 69 dshape[2], dshape[3])) 78 assert len(dshape) == 3 [all …]
|
H A D | graphpack.py | 43 int(dshape[2]), int(dshape[3]))) 60 assert len(dshape) == 4 66 int(dshape[2]), int(dshape[3]))) 75 dshape = _to_shape(dshape) 76 assert len(dshape) == 4 77 assert dshape[0] % cfactor == 0 78 assert dshape[1] % cfactor == 0 82 dshape[2], dshape[3])) 91 dshape = _to_shape(dshape) 92 assert len(dshape) == 3 [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/cudnn/ |
H A D | cudnn_softmax_activation-inl.h | 61 data = in_data.get_with_shape<gpu, 4, real_t>(dshape, s); in Forward() 67 Shape<4> dshape; in Forward() local 71 dshape[i] = in_data.shape_[i]; in Forward() 73 dshape[i] = 1; in Forward() 75 size_left /= dshape[i]; in Forward() 77 dshape[3] = size_left; in Forward() 126 Shape<4> dshape; in Backward() local 130 dshape[i] = in_grad.shape_[i]; in Backward() 132 dshape[i] = 1; in Backward() 134 size_left /= dshape[i]; in Backward() [all …]
|
H A D | cudnn_activation-inl.h | 79 Shape<4> dshape = Shape4(in_data.shape_[0], in Forward() local 84 Shape<4> dshape; in Forward() local 88 dshape[i] = in_data.shape_[i]; in Forward() 90 dshape[i] = 1; in Forward() 92 size_left /= dshape[i]; in Forward() 94 dshape[3] = size_left; in Forward() 141 Shape<4> dshape; in Backward() local 145 dshape[i] = in_grad.shape_[i]; in Backward() 147 dshape[i] = 1; in Backward() 149 size_left /= dshape[i]; in Backward() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/cudnn/ |
H A D | cudnn_softmax_activation-inl.h | 61 data = in_data.get_with_shape<gpu, 4, real_t>(dshape, s); in Forward() 67 Shape<4> dshape; in Forward() local 71 dshape[i] = in_data.shape_[i]; in Forward() 73 dshape[i] = 1; in Forward() 75 size_left /= dshape[i]; in Forward() 77 dshape[3] = size_left; in Forward() 126 Shape<4> dshape; in Backward() local 130 dshape[i] = in_grad.shape_[i]; in Backward() 132 dshape[i] = 1; in Backward() 134 size_left /= dshape[i]; in Backward() [all …]
|
H A D | cudnn_activation-inl.h | 79 Shape<4> dshape = Shape4(in_data.shape_[0], in Forward() local 84 Shape<4> dshape; in Forward() local 88 dshape[i] = in_data.shape_[i]; in Forward() 90 dshape[i] = 1; in Forward() 92 size_left /= dshape[i]; in Forward() 94 dshape[3] = size_left; in Forward() 141 Shape<4> dshape; in Backward() local 145 dshape[i] = in_grad.shape_[i]; in Backward() 147 dshape[i] = 1; in Backward() 149 size_left /= dshape[i]; in Backward() [all …]
|
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/relay/ |
H A D | test_pass_fuse_ops.py | 52 def before(dshape): argument 126 z = before(dshape) 149 p0 = relay.var("p0", shape=(dshape[0], dshape[1], dshape[2]//2, dshape[3]//2)) 163 z = before(dshape) 188 p0 = relay.var("p0", shape=(dshape[0], dshape[1], dshape[2]//2, dshape[3]//2)) 200 z = before(dshape) 233 dshape = (10, 20) 268 dshape = () 414 p01 = relay.var("p01", shape=(1, dshape[1]*9, dshape[2], dshape[3])) 420 p02 = relay.var("p02", shape=(1, dshape[1]*9, dshape[2]//2, dshape[3]//2)) [all …]
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/relay/ |
H A D | test_pass_fuse_ops.py | 52 def before(dshape): argument 126 z = before(dshape) 149 p0 = relay.var("p0", shape=(dshape[0], dshape[1], dshape[2]//2, dshape[3]//2)) 163 z = before(dshape) 188 p0 = relay.var("p0", shape=(dshape[0], dshape[1], dshape[2]//2, dshape[3]//2)) 200 z = before(dshape) 233 dshape = (10, 20) 268 dshape = () 414 p01 = relay.var("p01", shape=(1, dshape[1]*9, dshape[2], dshape[3])) 420 p02 = relay.var("p02", shape=(1, dshape[1]*9, dshape[2]//2, dshape[3]//2)) [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | convolution.cc | 105 mxnet::dim_size_is_known(dshape, 1) ? dshape[1] / param_.num_group : -1, in ConvolutionShape() 117 if (dshape[1] != -1) { in ConvolutionShape() 129 oshape[0] = dshape[0]; in ConvolutionShape() 138 dshape[0] = oshape[0]; in ConvolutionShape() 145 if (dshape[2] != -1) { in ConvolutionShape() 155 mxnet::dim_size_is_known(dshape, 1) ? dshape[1] / param_.num_group : -1, in ConvolutionShape() 168 if (dshape[1] != -1) { in ConvolutionShape() 180 oshape[0] = dshape[0]; in ConvolutionShape() 191 dshape[0] = oshape[0]; in ConvolutionShape() 214 mxnet::dim_size_is_known(dshape, 1) ? dshape[1] / param_.num_group : -1, in ConvolutionShape() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/nn/ |
H A D | convolution.cc | 105 mxnet::dim_size_is_known(dshape, 1) ? dshape[1] / param_.num_group : -1, in ConvolutionShape() 117 if (dshape[1] != -1) { in ConvolutionShape() 129 oshape[0] = dshape[0]; in ConvolutionShape() 138 dshape[0] = oshape[0]; in ConvolutionShape() 145 if (dshape[2] != -1) { in ConvolutionShape() 155 mxnet::dim_size_is_known(dshape, 1) ? dshape[1] / param_.num_group : -1, in ConvolutionShape() 168 if (dshape[1] != -1) { in ConvolutionShape() 180 oshape[0] = dshape[0]; in ConvolutionShape() 191 dshape[0] = oshape[0]; in ConvolutionShape() 214 mxnet::dim_size_is_known(dshape, 1) ? dshape[1] / param_.num_group : -1, in ConvolutionShape() [all …]
|
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | l2_normalization.cc | 53 Shape<2> dshape = Shape2(orig_shape[0], in Forward() local 56 .get_with_shape<cpu, 2, DType>(dshape, s); in Forward() 58 .get_with_shape<cpu, 2, DType>(dshape, s); in Forward() 73 Shape<3> dshape = Shape3(orig_shape[0], orig_shape[1], in Forward() local 76 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() 78 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() 79 Shape<2> norm_shape = Shape2(dshape[0], dshape[2]); in Forward() 97 Shape<3> dshape = Shape3(orig_shape[0], orig_shape[1], in Forward() local 100 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() 102 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() [all …]
|
H A D | l2_normalization-inl.h | 89 Shape<2> dshape = Shape2(orig_shape[0], in Forward() local 92 .get_with_shape<xpu, 2, DType>(dshape, s); in Forward() 111 Shape<2> norm_shape = Shape2(dshape[0], dshape[2]); in Forward() 129 Shape<2> norm_shape = Shape2(dshape[0], dshape[1]); in Forward() 160 Shape<2> dshape = Shape2(orig_shape[0], in Backward() local 185 Shape<2> norm_shape = Shape2(dshape[0], dshape[2]); in Backward() 204 Shape<2> norm_shape = Shape2(dshape[0], dshape[1]); in Backward() 271 out_shape->push_back(dshape); in InferShape() 273 out_shape->push_back(Shape1(dshape[0])); in InferShape() 276 mxnet::TShape norm_shape = dshape; in InferShape() [all …]
|
H A D | slice_channel-inl.h | 97 dshape, s); in Forward() 133 dshape, s); in Backward() 194 CHECK_LT(param_.axis, dshape.ndim()); in InferShape() 196 CHECK_LT(param_.axis + dshape.ndim(), dshape.ndim()); in InferShape() 200 real_axis += dshape.ndim(); in InferShape() 208 << dshape[real_axis]; in InferShape() 215 if (dshape[real_axis] >= 0) { in InferShape() 216 dshape[real_axis] /= param_.num_outputs; in InferShape() 221 dshape[d] = dshape[d+1]; in InferShape() 223 dshape = mxnet::TShape(&dshape[0], &dshape[dshape.ndim()-1]); in InferShape() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | l2_normalization.cc | 53 Shape<2> dshape = Shape2(orig_shape[0], in Forward() local 56 .get_with_shape<cpu, 2, DType>(dshape, s); in Forward() 58 .get_with_shape<cpu, 2, DType>(dshape, s); in Forward() 73 Shape<3> dshape = Shape3(orig_shape[0], orig_shape[1], in Forward() local 76 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() 78 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() 79 Shape<2> norm_shape = Shape2(dshape[0], dshape[2]); in Forward() 97 Shape<3> dshape = Shape3(orig_shape[0], orig_shape[1], in Forward() local 100 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() 102 .get_with_shape<cpu, 3, DType>(dshape, s); in Forward() [all …]
|
H A D | l2_normalization-inl.h | 89 Shape<2> dshape = Shape2(orig_shape[0], in Forward() local 92 .get_with_shape<xpu, 2, DType>(dshape, s); in Forward() 111 Shape<2> norm_shape = Shape2(dshape[0], dshape[2]); in Forward() 129 Shape<2> norm_shape = Shape2(dshape[0], dshape[1]); in Forward() 160 Shape<2> dshape = Shape2(orig_shape[0], in Backward() local 185 Shape<2> norm_shape = Shape2(dshape[0], dshape[2]); in Backward() 204 Shape<2> norm_shape = Shape2(dshape[0], dshape[1]); in Backward() 271 out_shape->push_back(dshape); in InferShape() 273 out_shape->push_back(Shape1(dshape[0])); in InferShape() 276 mxnet::TShape norm_shape = dshape; in InferShape() [all …]
|
H A D | slice_channel-inl.h | 97 dshape, s); in Forward() 133 dshape, s); in Backward() 194 CHECK_LT(param_.axis, dshape.ndim()); in InferShape() 196 CHECK_LT(param_.axis + dshape.ndim(), dshape.ndim()); in InferShape() 200 real_axis += dshape.ndim(); in InferShape() 208 << dshape[real_axis]; in InferShape() 215 if (dshape[real_axis] >= 0) { in InferShape() 216 dshape[real_axis] /= param_.num_outputs; in InferShape() 221 dshape[d] = dshape[d+1]; in InferShape() 223 dshape = mxnet::TShape(&dshape[0], &dshape[dshape.ndim()-1]); in InferShape() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/relay/ |
H A D | test_pass_fuse_ops.py | 56 def before(dshape): argument 112 z = before(dshape) 135 p0 = relay.var("p0", shape=(dshape[0], dshape[1], dshape[2] // 2, dshape[3] // 2)) 149 z = before(dshape) 174 p0 = relay.var("p0", shape=(dshape[0], dshape[1], dshape[2] // 2, dshape[3] // 2)) 186 z = before(dshape) 219 dshape = (10, 20) 251 dshape = () 398 p01 = relay.var("p01", shape=(1, dshape[1] * 9, dshape[2], dshape[3])) 404 p02 = relay.var("p02", shape=(1, dshape[1] * 9, dshape[2] // 2, dshape[3] // 2)) [all …]
|
/dports/graphics/panomatic/panomatic-0.9.4/vigra/include/vigra/ |
H A D | multi_pointoperators.hxx | 220 DestIterator dend = d + dshape[N]; in copyMultiArrayImpl() 407 DestIterator dend = d + dshape[0]; in transformMultiArrayReduceImpl() 425 DestIterator dend = d + dshape[N]; in transformMultiArrayReduceImpl() 445 vigra_precondition(dshape[i] == 1 || sshape[i] == dshape[i], in transformMultiArrayImpl() 449 if(dshape[i] != 1) in transformMultiArrayImpl() 482 DestIterator dend = d + dshape[N]; in transformMultiArrayExpandImpl() 785 DestIterator dend = d + dshape[0]; in combineTwoMultiArraysReduceImpl() 806 DestIterator dend = d + dshape[N]; in combineTwoMultiArraysReduceImpl() 831 (dshape[i] == 1 || sshape1[i] == dshape[i]), in combineTwoMultiArraysImpl() 836 if(dshape[i] != 1) in combineTwoMultiArraysImpl() [all …]
|
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/ |
H A D | convolution.cc | 81 dshape = ConvertLayout(dshape, in_layout, kNCHW); in Conv2DInferShape() 119 if (dshape[2] != 0) { in Conv2DInferShape() 122 if (dshape[3] != 0) { in Conv2DInferShape() 130 dshape[0] = oshape[0]; in Conv2DInferShape() 140 if (dshape[2] != 0) { in Conv2DInferShape() 144 if (dshape[3] != 0) { in Conv2DInferShape() 184 dshape = ConvertLayout(dshape, in_layout, kNCHW); in WinogradConv2DInferShape() 219 if (dshape[2] != 0) { in WinogradConv2DInferShape() 222 if (dshape[3] != 0) { in WinogradConv2DInferShape() 240 if (dshape[2] != 0) { in WinogradConv2DInferShape() [all …]
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/ |
H A D | convolution.cc | 81 dshape = ConvertLayout(dshape, in_layout, kNCHW); in Conv2DInferShape() 119 if (dshape[2] != 0) { in Conv2DInferShape() 122 if (dshape[3] != 0) { in Conv2DInferShape() 130 dshape[0] = oshape[0]; in Conv2DInferShape() 140 if (dshape[2] != 0) { in Conv2DInferShape() 144 if (dshape[3] != 0) { in Conv2DInferShape() 184 dshape = ConvertLayout(dshape, in_layout, kNCHW); in WinogradConv2DInferShape() 219 if (dshape[2] != 0) { in WinogradConv2DInferShape() 222 if (dshape[3] != 0) { in WinogradConv2DInferShape() 240 if (dshape[2] != 0) { in WinogradConv2DInferShape() [all …]
|