vinx13 commented on a change in pull request #8056:
URL: https://github.com/apache/tvm/pull/8056#discussion_r657327709
##########
File path: include/tvm/topi/nn.h
##########
@@ -642,6 +643,54 @@ inline tvm::te::Tensor batch_to_space_nd(const
tvm::te::Tensor& data,
out = strided_slice(out, begin_idx, end_idx, strides);
return out;
}
+
+/*!
+ * \brief Negative log likelihood loss.
+ *
+ * \param predictions The prediction tensor.
+ * \param targets The target tensor.
+ * \param weights A manual rescaling weight given to each class.
+ * \param reduction The reduction method to apply to the output.
+ * \param ignore_index The target value to ignore.
+ * \param name The name of the operation.
+ * \param tag The tag to mark the operation.
+ *
+ * \return A Tensor whose op member is the batch_to_space_nd operation
+ */
+inline Tensor nll_loss(const Tensor& predictions, const Tensor& targets, const
Tensor& weights,
+ std::string reduction = "mean", int ignore_index = -100,
+ const std::string name = "nll_loss", const std::string
tag = kBroadcast) {
Review comment:
tag here is topi-level, sometimes we use it to identify a specific
`compute` operation during schedule, otherwise we can leave it empty
--
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.
For queries about this service, please contact Infrastructure at:
[email protected]
