def verify_slice_like(np_data, np_shape_like, axis=[]): dtype = "float32" np_data = np_data.astype(dtype) np_shape_like = np_shape_like.astype(dtype) np_result = np_slice_like(np_data, np_shape_like, axis) data1 = sym.Variable("data1") data2 = sym.Variable("data2") net = sym.slice_like(data=data1, slice_like=data2, axis=axis) for target, ctx in ctx_list(): graph, lib, _ = nnvm.compiler.build(net, target, {"data1": np_data.shape, "data2": np_shape_like.shape}) m = graph_runtime.create(graph, lib, ctx) m.set_input(**{"data1": np_data, "data2": np_shape_like}) m.run() out = m.get_output(0, tvm.nd.empty(np_result.shape, dtype)) np.testing.assert_allclose(out.asnumpy(), np_result, atol=1e-5, rtol=1e-5)
def verify_slice_like(np_data, np_shape_like, axis=[]): dtype = "float32" np_data = np_data.astype(dtype) np_shape_like = np_shape_like.astype(dtype) np_result = np_slice_like(np_data, np_shape_like, axis) data1 = sym.Variable("data1") data2 = sym.Variable("data2") net = sym.slice_like(data=data1, slice_like=data2, axis=axis) for target, ctx in ctx_list(): graph, lib, _ = nnvm.compiler.build(net, target, {"data1": np_data.shape, "data2": np_shape_like.shape}) m = graph_runtime.create(graph, lib, ctx) m.set_input(**{"data1": np_data, "data2": np_shape_like}) m.run() out = m.get_output(0, tvm.nd.empty(np_result.shape, dtype)) tvm.testing.assert_allclose(out.asnumpy(), np_result, atol=1e-5, rtol=1e-5)