| 33 | |
| 34 | template <typename T, typename Context> |
| 35 | void SqrtKernel(const Context& dev_ctx, |
| 36 | const SelectedRows& x, |
| 37 | SelectedRows* out) { |
| 38 | out->set_rows(x.rows()); |
| 39 | out->set_height(x.height()); |
| 40 | phi::SqrtKernel<T, Context>(dev_ctx, x.value(), out->mutable_value()); |
| 41 | } |
| 42 | |
| 43 | } // namespace sr |
| 44 | } // namespace phi |
nothing calls this directly
no test coverage detected