| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/unittest/ |
| H A D | test_infer_shape.py | 160 def check(in_shape, out_shape, **kwargs): argument
193 out_shape, argument
213 def check(in_shape, out_shape, **kwargs): argument
244 def check(in_shape, out_shape, **kwargs): argument
274 def check(in_shape, out_shape, **kwargs): argument
289 def check(in_shape, tshape, out_shape): argument
313 def check(in_shape, axis, out_shape): argument
325 def check(in_shape, out_shape, **kwargs): argument
336 def check(in_shape, tshape, out_shape): argument
347 def check(lhs_shape, rhs_shape, out_shape): argument
[all …]
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/unittest/ |
| H A D | test_infer_shape.py | 160 def check(in_shape, out_shape, **kwargs): argument
193 out_shape, argument
213 def check(in_shape, out_shape, **kwargs): argument
244 def check(in_shape, out_shape, **kwargs): argument
274 def check(in_shape, out_shape, **kwargs): argument
289 def check(in_shape, tshape, out_shape): argument
313 def check(in_shape, axis, out_shape): argument
325 def check(in_shape, out_shape, **kwargs): argument
336 def check(in_shape, tshape, out_shape): argument
347 def check(lhs_shape, rhs_shape, out_shape): argument
[all …]
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/include/topi/image/ |
| H A D | resize.h | 98 Array<Expr> out_shape; variable
132 Array<Expr> out_shape; variable
166 Array<Expr> out_shape; variable
234 Array<Expr> out_shape; variable
311 Array<Expr> out_shape; variable
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/include/topi/image/ |
| H A D | resize.h | 98 Array<Expr> out_shape; variable
132 Array<Expr> out_shape; variable
166 Array<Expr> out_shape; variable
234 Array<Expr> out_shape; variable
311 Array<Expr> out_shape; variable
|
| /dports/science/py-chainer/chainer-7.8.0/chainerx_cc/chainerx/routines/ |
| H A D | sorting.cc | 23 Shape out_shape{}; in ArgMax() local
56 Shape out_shape{}; in ArgMin() local
95 Shape out_shape{}; in NanArgMax() local
129 Shape out_shape{}; in NanArgMin() local
|
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/compiler/ |
| H A D | fold_scale_axis.cc | 166 std::vector<TShape> in_shape, out_shape; in FoldScaleAxis() local
345 const std::vector<TShape>& out_shape, in ReluScaleAxisBackward()
355 const std::vector<TShape>& out_shape, in ReluScaleAxisForward()
379 const std::vector<TShape>& out_shape, in Pool2DBackward()
393 const std::vector<TShape>& out_shape, in Pool2DForward()
420 const std::vector<TShape>& out_shape, in BroadcastAddSubScaleAxisBackward()
444 const std::vector<TShape>& out_shape, in BroadcastAddSubScaleAxisForward()
481 const std::vector<TShape>& out_shape, in Conv2DScaleAxisBackward()
506 const std::vector<TShape>& out_shape, in Conv2DScaleAxisForward()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/compiler/ |
| H A D | fold_scale_axis.cc | 166 std::vector<TShape> in_shape, out_shape; in FoldScaleAxis() local
345 const std::vector<TShape>& out_shape, in ReluScaleAxisBackward()
355 const std::vector<TShape>& out_shape, in ReluScaleAxisForward()
379 const std::vector<TShape>& out_shape, in Pool2DBackward()
393 const std::vector<TShape>& out_shape, in Pool2DForward()
420 const std::vector<TShape>& out_shape, in BroadcastAddSubScaleAxisBackward()
444 const std::vector<TShape>& out_shape, in BroadcastAddSubScaleAxisForward()
481 const std::vector<TShape>& out_shape, in Conv2DScaleAxisBackward()
506 const std::vector<TShape>& out_shape, in Conv2DScaleAxisForward()
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/numpy/ |
| H A D | np_matmul_op.cc | 80 mxnet::TShape out_shape(a_ndim - 1, -1); in NumpyMatmulShape() local
97 mxnet::TShape out_shape(b_ndim - 1, -1); in NumpyMatmulShape() local
113 mxnet::TShape out_shape(ndim, -1); in NumpyMatmulShape() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/numpy/ |
| H A D | np_matmul_op.cc | 80 mxnet::TShape out_shape(a_ndim - 1, -1); in NumpyMatmulShape() local
97 mxnet::TShape out_shape(b_ndim - 1, -1); in NumpyMatmulShape() local
113 mxnet::TShape out_shape(ndim, -1); in NumpyMatmulShape() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/include/tvm/topi/ |
| H A D | transform.h | 342 Array<PrimExpr> out_shape; variable
400 Array<PrimExpr> out_shape; variable
451 Array<PrimExpr> out_shape; variable
611 Array<PrimExpr> out_shape; variable
712 Array<PrimExpr> out_shape = indices->shape; variable
763 Array<PrimExpr> out_shape = data->shape; variable
803 Array<PrimExpr> out_shape; variable
1092 Array<PrimExpr> out_shape; variable
1135 Array<PrimExpr> out_shape; variable
1320 Array<PrimExpr> out_shape; variable
|
| /dports/misc/tvm/incubator-tvm-0.6.1/topi/include/topi/ |
| H A D | transform.h | 262 Array<Expr> out_shape; variable
323 Array<Expr> out_shape; variable
378 Array<Expr> out_shape; variable
537 Array<Expr> out_shape; variable
638 Array<Expr> out_shape = indices->shape; variable
689 Array<Expr> out_shape = data->shape; variable
730 Array<Expr> out_shape; variable
975 Array<Expr> out_shape; variable
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/topi/include/topi/ |
| H A D | transform.h | 262 Array<Expr> out_shape; variable
323 Array<Expr> out_shape; variable
378 Array<Expr> out_shape; variable
537 Array<Expr> out_shape; variable
638 Array<Expr> out_shape = indices->shape; variable
689 Array<Expr> out_shape = data->shape; variable
730 Array<Expr> out_shape; variable
975 Array<Expr> out_shape; variable
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | deformable_convolution.cc | 20 mxnet::ShapeVector *out_shape, in CreateOp()
33 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
|
| H A D | modulated_deformable_convolution.cc | 18 std::vector<TShape> *out_shape, in CreateOp()
31 std::vector<TShape> out_shape, aux_shape; in CreateOperatorEx() local
|
| /dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
| H A D | convolution_v1.cc | 38 mxnet::ShapeVector *out_shape, in CreateOp()
51 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/contrib/ |
| H A D | modulated_deformable_convolution.cc | 18 std::vector<TShape> *out_shape, in CreateOp()
31 std::vector<TShape> out_shape, aux_shape; in CreateOperatorEx() local
|
| H A D | deformable_convolution.cc | 20 mxnet::ShapeVector *out_shape, in CreateOp()
33 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
| H A D | convolution_v1.cc | 38 mxnet::ShapeVector *out_shape, in CreateOp()
51 mxnet::ShapeVector out_shape, aux_shape; in CreateOperatorEx() local
|
| /dports/devel/py-pythran/pythran-0.11.0/pythran/pythonic/numpy/fft/ |
| H A D | c2c.hpp | 37 auto out_shape = sutils::getshape(in_array); in _copy_to_double() local
50 auto out_shape = sutils::getshape(in_array); in _copy_to_complex() local
64 auto out_shape = sutils::getshape(in_array); in _copy_to_complex() local
159 auto out_shape = sutils::getshape(in_array); in c2r() local
206 auto out_shape = sutils::getshape(in_array); in c2c() local
256 auto out_shape = sutils::getshape(in_array); in r2c() local
|
| /dports/misc/py-xgboost/xgboost-1.5.1/src/c_api/ |
| H A D | c_api.cu | 61 xgboost::bst_ulong const **out_shape, in InplacePreidctCuda()
95 DMatrixHandle m, xgboost::bst_ulong const **out_shape, in XGBoosterPredictFromCudaColumnar()
107 DMatrixHandle m, xgboost::bst_ulong const **out_shape, in XGBoosterPredictFromCudaArray()
|
| /dports/misc/xgboost/xgboost-1.5.1/src/c_api/ |
| H A D | c_api.cu | 61 xgboost::bst_ulong const **out_shape, in InplacePreidctCuda()
95 DMatrixHandle m, xgboost::bst_ulong const **out_shape, in XGBoosterPredictFromCudaColumnar()
107 DMatrixHandle m, xgboost::bst_ulong const **out_shape, in XGBoosterPredictFromCudaArray()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/ |
| H A D | convolution.cc | 51 std::vector<TShape>* out_shape) { in Conv2DInferShape()
154 std::vector<TShape>* out_shape) { in WinogradConv2DInferShape()
379 std::vector<TShape> *out_shape) { in __anona081f46b0202()
464 std::vector<TShape> *out_shape) { in __anona081f46b0402()
542 std::vector<TShape>* out_shape) { in Conv2DTransposeInferShape()
|
| /dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/src/top/nn/ |
| H A D | convolution.cc | 51 std::vector<TShape>* out_shape) { in Conv2DInferShape()
154 std::vector<TShape>* out_shape) { in WinogradConv2DInferShape()
379 std::vector<TShape> *out_shape) { in __anon26abcf010202()
464 std::vector<TShape> *out_shape) { in __anon26abcf010402()
542 std::vector<TShape>* out_shape) { in Conv2DTransposeInferShape()
|
| /dports/misc/tvm/incubator-tvm-0.6.1/tutorials/frontend/ |
| H A D | from_darknet.py | 146 out_shape = (layer_attr[0], layer_attr[1]//layer_attr[0], variable
162 out_shape = (layer_attr[0], layer_attr[1]//layer_attr[0], variable
176 out_shape = (layer_attr[0], layer_attr[1]//layer_attr[0], variable
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tutorials/frontend/ |
| H A D | from_darknet.py | 143 out_shape = (layer_attr[0], layer_attr[1] // layer_attr[0], layer_attr[2], layer_attr[3]) variable
158 out_shape = (layer_attr[0], layer_attr[1] // layer_attr[0], layer_attr[2], layer_attr[3]) variable
171 out_shape = (layer_attr[0], layer_attr[1] // layer_attr[0], layer_attr[2], layer_attr[3]) variable
|