[
https://issues.apache.org/jira/browse/SYSTEMML-1389?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Mike Dusenberry updated SYSTEMML-1389:
--------------------------------------
Summary: Update API: Pass in all outputs from `forward` to `backward` for
performance (was: Update API: Pass In All Outputs From `forward` To `backward`
for Performance)
> Update API: Pass in all outputs from `forward` to `backward` for performance
> ----------------------------------------------------------------------------
>
> Key: SYSTEMML-1389
> URL: https://issues.apache.org/jira/browse/SYSTEMML-1389
> Project: SystemML
> Issue Type: Sub-task
> Reporter: Mike Dusenberry
> Assignee: Mike Dusenberry
>
> Currently, we do not pass the outputs of the {{forward}} functions to the
> {{backward}} functions in the {{nn}} library. Many of the layers do not need
> the outputs, but a few current ones, and possible future ones, could benefit
> from having access to the outputs. This aims to update the {{backward}} API
> to include (1) all upstream derivatives, (2) all outputs from {{forward}},
> (3) all inputs to {{forward}}. Effectively, this would be equivalent to
> having a class object that maintains all configuration and input + output
> tensors.
> cc [~niketanpansare] This will require changes to the Barista API, but I
> believe it will be beneficial for performance.
--
This message was sent by Atlassian JIRA
(v6.3.15#6346)
