| 122 | } |
| 123 | |
| 124 | std::unique_ptr<OpKernel> GetDiv(DataType dtype, const DeviceType& device_type, |
| 125 | DeviceBase* device) { |
| 126 | NodeDef node_def; |
| 127 | NodeDefBuilder builder("add_node", "Div"); |
| 128 | TF_CHECK_OK(builder.Attr("T", dtype) |
| 129 | .Input(FakeInput(dtype)) |
| 130 | .Input(FakeInput(dtype)) |
| 131 | .Finalize(&node_def)); |
| 132 | return GetKernel(node_def, device_type, device); |
| 133 | } |
| 134 | |
| 135 | static int64 kStepId = 123; |
| 136 |