Home
last modified time | relevance | path

Searched refs:dshape (Results 1 – 25 of 506) sorted by relevance

12345678910>>...21

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/vta/python/vta/top/
H A Dgraphpack.py51 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 Dbilinear_resize-inl.h284 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 Dbilinear_resize-inl.h284 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 Dnnvm_graphpack.py26 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 Dgraphpack.py43 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 Dnnvm_graphpack.py26 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 Dgraphpack.py43 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 Dcudnn_softmax_activation-inl.h61 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 Dcudnn_activation-inl.h79 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 Dcudnn_softmax_activation-inl.h61 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 Dcudnn_activation-inl.h79 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 Dtest_pass_fuse_ops.py52 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 Dtest_pass_fuse_ops.py52 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 Dconvolution.cc105 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 Dconvolution.cc105 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 Dl2_normalization.cc53 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 Dl2_normalization-inl.h89 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 Dslice_channel-inl.h97 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 Dl2_normalization.cc53 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 Dl2_normalization-inl.h89 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 Dslice_channel-inl.h97 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 Dtest_pass_fuse_ops.py56 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 Dmulti_pointoperators.hxx220 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 Dconvolution.cc81 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 Dconvolution.cc81 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 …]

12345678910>>...21