| 135 | } |
| 136 | |
| 137 | Node* Reduce(Graph* g, const string& reduce, Node* data, Node* axes, |
| 138 | bool keep_dims) { |
| 139 | Node* ret; |
| 140 | TF_CHECK_OK(NodeBuilder(g->NewName("n"), reduce, g->op_registry()) |
| 141 | .Input(data) |
| 142 | .Input(axes) |
| 143 | .Attr("keep_dims", keep_dims) |
| 144 | .Finalize(g, &ret)); |
| 145 | return ret; |
| 146 | } |
| 147 | |
| 148 | Node* QuantizeToUINT8(Graph* g, Node* data) { |
| 149 | Node* ret; |
no test coverage detected