| 123 | } |
| 124 | |
| 125 | Status Softmax::AddToQueue(CLCommandQueue* queue) { |
| 126 | RETURN_IF_ERROR(BindArguments()); |
| 127 | return queue->DispatchImplicit(kernel_, GetGridSize(), work_group_size_); |
| 128 | } |
| 129 | |
| 130 | Softmax CreateSoftmax(const OperationDef& definition) { |
| 131 | return Softmax(definition); |