[TVM Discuss] [Development] Want help adding Batch_Matmul op for tenforflow frontend

2019-07-22 Thread Yong Wu via TVM Discuss
@Shawhey, not sure why you modified topi and relay op, is it to add broadcasting? seems there is no need to touch topi and relay op because batch_matmul was added in topi and relay a few months ago. To support tf BatchMatMul, I think we should do is to add the support in python/tvm/relay/fron

[TVM Discuss] [Development] Want help adding Batch_Matmul op for tenforflow frontend

2019-07-21 Thread Shawhey via TVM Discuss
https://github.com/dmlc/tvm/issues/3414 I am trying to add batch_mamtul op for tensorflow frontend. I edited the files below: python/tvm/relay/frontend/tensorflow.py, include/tvm/relay/attrs/nn.h, python/tvm/relay/op/nn/_nn.py, python/tvm/relay/op/nn/nn.py, python/tvm/relay/op/op_attrs.py