You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/09/01 06:01:57 UTC

[GitHub] [tvm] liaojianjin opened a new pull request #8894: [ONNX] Add index_put operator

liaojianjin opened a new pull request #8894:
URL: https://github.com/apache/tvm/pull/8894


   Add `aten: index_put` operator support to ONNX frontend
   Operator description:
   https://pytorch.org/docs/stable/generated/torch.Tensor.index_put_.html#torch.Tensor.index_put_
   In the process of converting the index_put operator of pytorch to onnx, the index of the slice type will be converted to a multi-dimensional tensor for storage.[ReshapeToAdvancedIndexingFormat](https://github.com/pytorch/pytorch/blob/f7743e92bf7e8655a5a712574b3608b0c8659d29/torch/csrc/jit/passes/onnx/pattern_conversion/pattern_conversion.cpp#L178)
   By expanding the slice into a 1-dimensional tensor, to indicate the location of the assignment value.


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] masahi commented on pull request #8894: [ONNX] Add index_put operator

Posted by GitBox <gi...@apache.org>.
masahi commented on pull request #8894:
URL: https://github.com/apache/tvm/pull/8894#issuecomment-912364324


   Thanks @liaojianjin 


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org

[GitHub] [tvm] masahi merged pull request #8894: [ONNX] Add index_put operator

Posted by GitBox <gi...@apache.org>.
masahi merged pull request #8894:
URL: https://github.com/apache/tvm/pull/8894


   


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org