I have done these:

    extern_mod = relay.transform.AnnotateTarget(['nuc_fpga'])(mod)
    extern_mod = relay.transform.MergeCompilerRegions()(extern_mod)
    extern_mod = relay.transform.PartitionGraph()(extern_mod)
    print("extern_mod:", extern_mod)


output is 

    %0 = nn.conv2d(%input0, %conv1.0.weight, strides=[2, 2], padding=[3, 3, 3, 
3], channels=64, kernel_size=[7, 7]) /* ty=Tensor[(1, 64, 112, 112), float32] 
*/;
    %1 = nn.bias_add(%0, %conv1.0.bias) /* ty=Tensor[(1, 64, 112, 112), 
float32] */;
    %2 = nn.relu(%1) /* ty=Tensor[(1, 64, 112, 112), float32] */;

 The partitioned graph still is nn.con2d





---
[Visit 
Topic](https://discuss.tvm.apache.org/t/question-byoc-replace-nn-conv2d-to-our-nucfpga-conv2d/9950/6)
 to respond.

You are receiving this because you enabled mailing list mode.

To unsubscribe from these emails, [click 
here](https://discuss.tvm.apache.org/email/unsubscribe/3bc7a785f0aae6d8e75736c9e79df8fd4d119154a93e6fa2ba827f5740fa2e3d).

Reply via email to