Home
last modified time | relevance | path

Searched refs:ishapes (Results 1 – 22 of 22) sorted by relevance

/dports/devel/py-numba/numba-0.51.2/numba/cuda/tests/cudapy/
H A Dtest_gufunc_scheduling.py20 ishapes=[(4, 5), (5, 7)],
31 ishapes=[(4, 5), (5, 7)],
42 ishapes=[(4, 5), (5, 7)],
53 ishapes=[(4, 5), (5, 7)],
64 ishapes=[(5,), (5,)],
75 ishapes=[(), ()],
86 ishapes=[(), ()],
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/caffe2/
H A Dtest_graph.py27 ishapes = {input_name: ishape}
28 graph_attr.set_shape_inputs(g1, ishapes)
29 graph_attr.set_shape_inputs(g2, ishapes)
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/caffe2/
H A Dtest_graph.py27 ishapes = {input_name: ishape}
28 graph_attr.set_shape_inputs(g1, ishapes)
29 graph_attr.set_shape_inputs(g2, ishapes)
/dports/misc/tvm/incubator-tvm-0.6.1/rust/runtime/tests/
H A Dbuild_model.py43 ishapes, _ = graph_util.infer_shape(graph, **input_shapes)
44 param_shapes = dict(zip(graph.index.input_names, ishapes))
/dports/misc/py-tvm/incubator-tvm-0.6.1/rust/runtime/tests/
H A Dbuild_model.py43 ishapes, _ = graph_util.infer_shape(graph, **input_shapes)
44 param_shapes = dict(zip(graph.index.input_names, ishapes))
/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/compiler/
H A Dtest_optimizer.py28 ishapes = {}
32 ishapes.update({name: shape})
38 graph, lib, rt_params = nnvm.compiler.build(symbol, target, shape=ishapes)
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/compiler/
H A Dtest_optimizer.py28 ishapes = {}
32 ishapes.update({name: shape})
38 graph, lib, rt_params = nnvm.compiler.build(symbol, target, shape=ishapes)
/dports/misc/tvm/incubator-tvm-0.6.1/rust/runtime/tests/test_nnvm/src/
H A Dbuild_test_graph.py43 ishapes, _ = graph_util.infer_shape(graph, **input_shapes)
44 param_shapes = dict(zip(graph.index.input_names, ishapes))
/dports/misc/py-tvm/incubator-tvm-0.6.1/rust/runtime/tests/test_nnvm/src/
H A Dbuild_test_graph.py43 ishapes, _ = graph_util.infer_shape(graph, **input_shapes)
44 param_shapes = dict(zip(graph.index.input_names, ishapes))
/dports/devel/py-numba/numba-0.51.2/numba/np/ufunc/
H A Ddeviceufunc.py559 def schedule(self, ishapes): argument
560 if len(ishapes) != self.nin:
568 for argn, (shape, symbols) in enumerate(zip(ishapes, self.sin)):
618 def __init__(self, parent, ishapes, oshapes, loopdims, pinned): argument
621 self.ishapes = ishapes
702 for p, cs in zip(params, schedule.ishapes):
/dports/math/py-theano/Theano-1.0.5/theano/tensor/
H A Dsubtensor.py1779 def infer_shape(self, node, ishapes): argument
1780 x, ilist = ishapes
2024 def infer_shape(self, node, ishapes): argument
2025 x, y, ilist = ishapes
2190 def infer_shape(self, node, ishapes): argument
2229 def infer_shape(self, node, ishapes): argument
2231 if len(ishapes) == 3:
2232 xshp, ind1shp, ind2shp = ishapes
2243 return super(AdvancedSubtensor, self).infer_shape(node, ishapes)
2343 def infer_shape(self, node, ishapes): argument
[all …]
H A Dbasic.py4430 def infer_shape(self, node, ishapes): argument
4433 assert len(ishapes) > 1
4434 n_dim = len(ishapes[1])
4435 for shp in ishapes[1:]:
4458 t_side = ishapes[1][dim]
4459 f_side = ishapes[1][dim]
4461 for shp in ishapes[2:]:
4942 def infer_shape(self, node, ishapes): argument
4965 if len(ishapes[0]) == 0:
4969 input_size = mul(*ishapes[0])
[all …]
H A Dopt.py810 def infer_shape(self, node, ishapes): argument
811 return [(len(ishapes),)]
/dports/math/py-theano/Theano-1.0.5/theano/compile/
H A Dops.py734 def infer_shape(self, node, ishapes): argument
735 assert len(ishapes) == 1
738 for ax in xrange(len(ishapes[0])):
742 l.append(ishapes[0][ax])
/dports/math/py-theano/Theano-1.0.5/theano/tensor/tests/
H A Dtest_blas.py546 def just_gemm(i, o, ishapes=[(4, 3), (3, 5), (4, 5), (), ()], argument
576 for sh in ishapes])
579 for sh in ishapes])
609 ishapes=[(5, 3), (3, 4), (4, 5), (), ()])
652 for sh in ishapes])
655 for sh in ishapes])
771 ishapes=[(2, 3), (3, 4), (2, 4), (2, 3), (3, 4),
777 ishapes=[(2, 3), (3, 4), (2, 4), (2, 3), (3, 4),
783 ishapes=[(2, 3), (3, 4), (2, 4), (2, 3), (3, 4),
808 ishapes = [(4, 3), (3, 5), (4, 5), (), (), (5, )]
[all …]
/dports/graphics/djvulibre/djvulibre-3.5.28/libdjvu/
H A DJB2Image.cpp1073 int ishapes = jim.get_inherited_shape_count(); in code() local
1074 jim.boxes.resize(0, nshapes-ishapes-1); in code()
1075 for (int i = ishapes; i < nshapes; i++) in code()
1076 jim.boxes[i-ishapes] = libinfo[i]; in code()
/dports/misc/py-mxnet/incubator-mxnet-1.9.0/tests/cpp/include/
H A Dtest_legacy_op.h169 mxnet::ShapeVector ishapes; in initBackward() local
170 allocateResources(opProp.BackwardResource(ishapes)); in initBackward()
/dports/misc/mxnet/incubator-mxnet-1.9.0/tests/cpp/include/
H A Dtest_legacy_op.h169 mxnet::ShapeVector ishapes; in initBackward() local
170 allocateResources(opProp.BackwardResource(ishapes)); in initBackward()
/dports/science/py-scipy/scipy-1.7.1/scipy/signal/tests/
H A Dtest_signaltools.py797 return [ishapes+(imode,) for ishapes, imode in product(shapes, modes)
799 (ishapes[0] > ishapes[1] and ishapes[2] > ishapes[3]) or
800 (ishapes[0] < ishapes[1] and ishapes[2] < ishapes[3])]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/pytorch/
H A Dtest_forward.py1892 def verify_script_model(pt_model, ishapes, targets): argument
1894 verify_model_vm(script_module, ishapes, targets=targets)
1899 ishapes = [data.shape for data in idata]
1900 verify_model_vm(traced_model, ishapes, idata=idata, targets=targets)
1903 def verify_model_vm(input_model, ishapes, idtype=torch.float, idata=None, targets=["llvm"]): argument
1904 input_names = ["i{}".format(idx) for idx, ish in enumerate(ishapes)]
1905 input_shapes = list(zip(input_names, ishapes))
1906 input_data = idata if idata else [torch.randn(shape, dtype=idtype) for shape in ishapes]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/relay/frontend/
H A Dpytorch.py2212 ishapes = _infer_shape(data)
2213 if dim >= len(ishapes):
2217 selections = ishapes[dim]
/dports/math/py-theano/Theano-1.0.5/theano/sparse/
H A Dbasic.py4265 def infer_shape(self, node, ishapes): argument