Home
last modified time | relevance | path

Searched refs:input2_dim (Results 1 – 5 of 5) sorted by relevance

/dports/misc/tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/coreml/
H A Dtest_forward.py165 def verify_ConcatLayerParams(input1_dim, input2_dim): argument
169 a_np2 = np.random.uniform(size=input2_dim).astype(dtype)
173 ('input2', datatypes.Array(*input2_dim))]
/dports/misc/py-tvm/incubator-tvm-0.6.1/nnvm/tests/python/frontend/coreml/
H A Dtest_forward.py165 def verify_ConcatLayerParams(input1_dim, input2_dim): argument
169 a_np2 = np.random.uniform(size=input2_dim).astype(dtype)
173 ('input2', datatypes.Array(*input2_dim))]
/dports/misc/py-tvm/incubator-tvm-0.6.1/tests/python/frontend/coreml/
H A Dtest_forward.py159 def verify_ConcatLayerParams(input1_dim, input2_dim): argument
163 a_np2 = np.random.uniform(size=input2_dim).astype(dtype)
167 ('input2', datatypes.Array(*input2_dim))]
/dports/misc/tvm/incubator-tvm-0.6.1/tests/python/frontend/coreml/
H A Dtest_forward.py159 def verify_ConcatLayerParams(input1_dim, input2_dim): argument
163 a_np2 = np.random.uniform(size=input2_dim).astype(dtype)
167 ('input2', datatypes.Array(*input2_dim))]
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/frontend/coreml/
H A Dtest_forward.py178 def verify_ConcatLayerParams(input1_dim, input2_dim): argument
182 a_np2 = np.random.uniform(size=input2_dim).astype(dtype)
185 inputs = [("input1", datatypes.Array(*input1_dim)), ("input2", datatypes.Array(*input2_dim))]