/dports/devel/py-numba/numba-0.51.2/numba/cuda/tests/cudapy/ |
H A D | test_gufunc_scheduling.py | 20 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 D | test_graph.py | 27 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 D | test_graph.py | 27 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 D | build_model.py | 43 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 D | build_model.py | 43 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 D | test_optimizer.py | 28 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 D | test_optimizer.py | 28 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 D | build_test_graph.py | 43 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 D | build_test_graph.py | 43 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 D | deviceufunc.py | 559 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 D | subtensor.py | 1779 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 D | basic.py | 4430 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 D | opt.py | 810 def infer_shape(self, node, ishapes): argument 811 return [(len(ishapes),)]
|
/dports/math/py-theano/Theano-1.0.5/theano/compile/ |
H A D | ops.py | 734 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 D | test_blas.py | 546 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 D | JB2Image.cpp | 1073 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 D | test_legacy_op.h | 169 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 D | test_legacy_op.h | 169 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 D | test_signaltools.py | 797 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 D | test_forward.py | 1892 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 D | pytorch.py | 2212 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 D | basic.py | 4265 def infer_shape(self, node, ishapes): argument
|