| 38 | // Two inputs and one output, all the same type. |
| 39 | template <class T> |
| 40 | class BinaryOp : public OpKernel { |
| 41 | public: |
| 42 | explicit BinaryOp(OpKernelConstruction* context) : OpKernel(context) { |
| 43 | const DataType dt = DataTypeToEnum<T>::v(); |
| 44 | OP_REQUIRES_OK(context, context->MatchSignature({dt, dt}, {dt})); |
| 45 | } |
| 46 | }; |
| 47 | |
| 48 | // For operations where the input and output are the same shape. |
| 49 | // |
no outgoing calls