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 D | test_forward.py | 165 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 D | test_forward.py | 165 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 D | test_forward.py | 159 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 D | test_forward.py | 159 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 D | test_forward.py | 178 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))]
|