Home
last modified time | relevance | path

Searched defs:output_shape (Results 1 – 25 of 160) sorted by relevance

1234567

/dports/misc/tvm/incubator-tvm-0.6.1/topi/python/topi/
H A Dargwhere.py23 def hybrid_argwhere_1d(output_shape, condition): argument
46 def hybrid_argwhere_2d(output_shape, condition): argument
72 def hybrid_argwhere_3d(output_shape, condition): argument
101 def hybrid_argwhere_4d(output_shape, condition): argument
133 def hybrid_argwhere_5d(output_shape, condition): argument
168 def argwhere(output_shape, condition): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/
H A Dargwhere.py23 def hybrid_argwhere_1d(output_shape, condition): argument
47 def hybrid_argwhere_2d(output_shape, condition): argument
74 def hybrid_argwhere_3d(output_shape, condition): argument
104 def hybrid_argwhere_4d(output_shape, condition): argument
137 def hybrid_argwhere_5d(output_shape, condition): argument
172 def argwhere(output_shape, condition): argument
/dports/misc/py-tvm/incubator-tvm-0.6.1/topi/python/topi/
H A Dargwhere.py23 def hybrid_argwhere_1d(output_shape, condition): argument
46 def hybrid_argwhere_2d(output_shape, condition): argument
72 def hybrid_argwhere_3d(output_shape, condition): argument
101 def hybrid_argwhere_4d(output_shape, condition): argument
133 def hybrid_argwhere_5d(output_shape, condition): argument
168 def argwhere(output_shape, condition): argument
/dports/misc/py-onnx/onnx-1.10.2/onnx/defs/tensor/
H A Dutils.cc12 TensorShapeProto* output_shape) { in resizeShapeInferenceHelper()
25 TensorShapeProto* output_shape) { in resizeShapeInferenceHelper()
57 auto* output_shape = getOutputShape(ctx, 0); in resizeShapeInference() local
107 TensorShapeProto* output_shape) { in resizeShapeInferenceHelper_opset7_to_10()
139 auto* output_shape = getOutputShape(ctx, 0); in resizeShapeInference_opset7_to_10() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/test_arm_compute_lib/
H A Dtest_reshape.py34 def _get_model(input_shape, output_shape, dtype, var_names): argument
41 def _get_expected_codegen(input_shape, output_shape, dtype): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/topi/nn/
H A Dupsampling.py30 output_shape=None, argument
113 output_shape=None, argument
/dports/misc/py-onnx/onnx-1.10.2/onnx/defs/reduction/
H A Dold.cc71 auto output_shape = in ReduceDocGenerator_opset12() local
201 auto output_shape = in ArgReduceDocGenerator_opset12() local
294 auto output_shape = in ReduceDocGenerator_opset1() local
422 auto output_shape = in ArgReduceDocGenerator_opset1() local
502 auto output_shape = in ArgReduceDocGenerator_opset11() local
H A Ddefs.cc138 auto output_shape = in ReduceDocGenerator() local
275 auto output_shape = in ArgReduceDocGenerator() local
/dports/graphics/py-scikit-image/scikit-image-0.19.0/skimage/transform/
H A D_warps.py24 def _preprocess_resize_output_shape(image, output_shape): argument
71 def resize(image, output_shape, order=None, mode='reflect', cval=0, clip=True, argument
527 output_shape=None, order=None, mode='reflect', cval=0, clip=True, argument
734 def warp(image, inverse_map, map_args={}, output_shape=None, order=None, argument
1053 def warp_polar(image, center=None, *, radius=None, output_shape=None, argument
1206 def resize_local_mean(image, output_shape, grid_mode=True, argument
/dports/misc/py-onnx-tf/onnx-tf-1.6.0/onnx_tf/handlers/backend/
H A Dunpool_mixin.py76 def _get_pads_from_output_shape(cls, unpool, output_shape): argument
125 def _unpool(cls, pool, ind, output_shape, scope='unpool'): argument
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/array/
H A Dspatial_transformer_grid.py18 def __init__(self, output_shape): argument
115 def spatial_transformer_grid(theta, output_shape, **kwargs): argument
H A Dresize_images.py204 def __init__(self, output_shape, mode, align_corners): argument
253 def __init__(self, input_shape, output_shape, mode, align_corners): argument
302 def resize_images(x, output_shape, *, mode='bilinear', align_corners=True): argument
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/contrib/tf_op/
H A Dmodule.py30 def func(self, name, output_dtype=None, output_shape=None): argument
55 def __init__(self, lib_path, func_name, output_dtype, output_shape): argument
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/defs/reduction/
H A Ddefs.cc50 auto output_shape = in ReduceDocGenerator() local
167 auto output_shape = in ArgReduceDocGenerator() local
/dports/misc/py-onnx/onnx-1.10.2/onnx/defs/generator/
H A Dold.cc172 auto* output_shape = getOutputShape(ctx, 0); in __anon980a70580102() local
296 auto* output_shape = getOutputShape(ctx, 0); in __anon980a70580402() local
/dports/graphics/py-scikit-image/scikit-image-0.19.0/skimage/util/
H A D_label.py6 def label_points(coords, output_shape): argument
/dports/misc/py-onnx/onnx-1.10.2/onnx/defs/nn/
H A Ddefs.cc148 auto output_shape = in convPoolShapeInference() local
437 auto& output_shape = getInputShape(ctx, 2); in maxUnpoolShapeInference() local
686 auto output_shape = in roiPoolTypeShapeInference() local
1256 std::vector<int64_t> output_shape; in convTransposeShapeInference() local
1465 auto output_shape = in globalPoolTypeShapeInference() local
2323 TensorShapeProto output_shape; in __anon4e3f06301702() local
2388 TensorShapeProto output_shape; in __anon4e3f06301902() local
/dports/misc/py-onnx/onnx-1.10.2/onnx/backend/test/case/model/
H A Dstringnormalizer.py22 …def make_graph(node, input_shape, output_shape): # type: (onnx.helper.NodeProto, Sequence[int], S… argument
H A Dexpand.py21 …def make_graph(node, input_shape, shape_shape, output_shape): # type: (onnx.helper.NodeProto, Seq… argument
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/backend/test/case/model/
H A Dstringnormalizer.py20 …def make_graph(node, input_shape, output_shape): # type: (onnx.helper.NodeProto, Sequence[int], S… argument
H A Dexpand.py19 …def make_graph(node, input_shape, shape_shape, output_shape): # type: (onnx.helper.NodeProto, Seq… argument
/dports/math/py-keras/Keras-2.4.3/tests/keras/layers/
H A Dcore_test.py189 def output_shape(input_shape): function
219 def output_shape(input_shape): function
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/test_ethosn/
H A Dtest_reshape.py27 def _get_model(input_shape, output_shape, dtype): argument
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/defs/nn/
H A Ddefs.cc131 auto output_shape = in convPoolShapeInference() local
530 auto& output_shape = getInputShape(ctx, 2); in maxUnpoolShapeInference() local
756 auto output_shape = in roiPoolTypeShapeInference() local
1297 std::vector<int64_t> output_shape; in convTransposeShapeInference() local
1482 auto output_shape = in gloablPoolTypeShapeInference() local
2040 TensorShapeProto output_shape; in __anon1663b9f71902() local
2105 TensorShapeProto output_shape; in __anon1663b9f71b02() local
/dports/misc/glow/glow-f24d960e3cc80db95ac0bc17b1900dbf60ca044a/thirdparty/onnx/onnx/defs/tensor/
H A Dold.cc386 auto* output_shape = getOutputShape(ctx, 0); in __anonfd5563130302() local
434 auto* output_shape = getOutputShape(ctx, 0); in __anonfd5563130402() local

1234567