Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Update operator.md
Browse files Browse the repository at this point in the history
  • Loading branch information
jermainewang committed Sep 19, 2015
1 parent 69ac3e1 commit 2657094
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions doc/developer-guide/operator.md
Original file line number Diff line number Diff line change
Expand Up @@ -70,8 +70,10 @@ It is possible that one convolution has several implementations and users want t
* **Request Resources:** Operation like `cudnnConvolutionForward` need a workspace to help computation. It is would be nice if the system could manage that since the system then could do optimizations like reuse the space and so on. MXNet defines two interfaces to achieve this:
```c++
virtual std::vector<ResourceRequest> ForwardResource(const std::vector<TShape> &in_shape) const;
virtual std::vector<ResourceRequest> BackwardResource(const std::vector<TShape> &in_shape) const;
virtual std::vector<ResourceRequest> ForwardResource(
const std::vector<TShape> &in_shape) const;
virtual std::vector<ResourceRequest> BackwardResource(
const std::vector<TShape> &in_shape) const;
```
The `ResourceRequest` structure (in `resource.h`) currently contains only a type flag:
```c++
Expand Down

0 comments on commit 2657094

Please sign in to comment.