MIOpen
MIOpen copied to clipboard
ROCmSoftwarePlatform
AMD's Machine Intelligence Library
* Added [SparseSoftmaxCrossEntropyWithLogits](https://www.tensorflow.org/api_docs/python/tf/nn/sparse_softmax_cross_entropy_with_logits) forward and backward. * Added driver test and gtest for SparseSoftmaxCrossEntropyWithLogits. * New API is guarded by MIOPEN_BETA_API macro. * Average over all cases: * SparseSoftmaxCrossEntropyWithLogits |...
* Added [MatrixBandPart](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/matrix-band-part). * Added driver test and gtest for MatrixBandPart. * New API is guarded by MIOPEN_BETA_API macro. * Average over all cases: * MatrixBandPart | Type | Forward...
* Added [Amax](https://pytorch.org/docs/stable/generated/torch.amax.html) & [Amin](https://pytorch.org/docs/stable/generated/torch.amin.html) backward because only backward has better performance over Rocm. * Added driver test and gtest for Amax & Amin. * New API is guarded by...
[ResourceApplyAdam ](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/resource-apply-adam) * Base on Adam op * Added Nesterov mode * Fixed bug in kernel code * This newly added mode does not affect performance.
* Added [ResourceApplyGradientDescent](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/resource-apply-gradient-descent). * Added driver test and gtest for ResourceApplyGradientDescent. * New API is guarded by MIOPEN_BETA_API macro. * Average over all cases: * ResourceApplyGradientDescent | Type | Forward...
* Added [ResourceApplyKerasMomentum](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/resource-apply-keras-momentum). * Added driver test and gtest for ResourceApplyKerasMomentum. * New API is guarded by MIOPEN_BETA_API macro. * Average over all cases: * ResourceApplyKerasMomentum | Type | Forward...
* Added [CartesianProd ](https://pytorch.org/docs/stable/generated/torch.cartesian_prod.html)forward and backward. * Added driver test and gtest for CartesianProd . * New API is guarded by MIOPEN_BETA_API macro. * Average over all cases: * CartesianProd...
* Added [Allclose](https://pytorch.org/docs/main/generated/torch.allclose.html). * Added driver test and gtest for Allclose. * New API is guarded by MIOPEN_BETA_API macro. * Average over all cases: * Allclose | Type | Allclose...
* Added [UnsortedSegmentSum](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/unsorted-segment-sum) forward and backward. * Added driver test and gtest for UnsortedSegmentSum. * New API is guarded by MIOPEN_BETA_API macro. * Average over all cases: * UnsortedSegmentSum |...
* Added contiguous [SoftmaxCrossEntropyWithLogits](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/softmax-cross-entropy-with-logits) forward and backward contiguous operation and kernel. * Added driver test and gtest for SoftmaxCrossEntropyWithLogits . * New API is guarded by MIOPEN_BETA_API macro. * Average...