| 111 | } |
| 112 | |
| 113 | std::unique_ptr<OpKernel> GetAdd(DataType dtype, const DeviceType& device_type, |
| 114 | DeviceBase* device) { |
| 115 | NodeDef node_def; |
| 116 | NodeDefBuilder builder("add_node", "Add"); |
| 117 | TF_CHECK_OK(builder.Attr("T", dtype) |
| 118 | .Input(FakeInput(dtype)) |
| 119 | .Input(FakeInput(dtype)) |
| 120 | .Finalize(&node_def)); |
| 121 | return GetKernel(node_def, device_type, device); |
| 122 | } |
| 123 | |
| 124 | std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, |
| 125 | DeviceBase* device) { |