replace BatchMatMul with default MatMul when convert pb to uff file

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano
1

Hi

There are some BatchMatMul operations in my tensorflow model(pb file) and i found that it is same as MatMul. So how do i replace BatchMatMul with default MatMul when convert pb to uff file ?

For example, if the operation of network/attention_block/MatMul is BatchMatMul, how do i replace it with default MatMul ?

"network/attention_block/MatMul": gs.create_plugin_node(name="ab_matmul", op=" ? ")

Thanks

2

Hi,

Like this:

matmul = gs.create_node(
    "BatchMatMul",
    op="Mul",
)

Thanks.