| 124 | } |
| 125 | |
| 126 | Node* Cumsum(Graph* g, Node* data, Node* axes, bool exclusive, bool reverse) { |
| 127 | Node* ret; |
| 128 | TF_CHECK_OK(NodeBuilder(g->NewName("n"), "Cumsum") |
| 129 | .Input(data) |
| 130 | .Input(axes) |
| 131 | .Attr("exclusive", exclusive) |
| 132 | .Attr("reverse", reverse) |
| 133 | .Finalize(g, &ret)); |
| 134 | return ret; |
| 135 | } |
| 136 | |
| 137 | Node* Reduce(Graph* g, const string& reduce, Node* data, Node* axes, |
| 138 | bool keep_dims) { |
no test coverage detected