MCPcopy Create free account
hub / github.com/PaddlePaddle/Paddle / SetSelectedRowsKernelOutput

Function SetSelectedRowsKernelOutput

paddle/phi/api/lib/api_gen_utils.cc:260–267  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

258}
259
260phi::SelectedRows* SetSelectedRowsKernelOutput(Tensor* out) {
261 if (!out->initialized()) {
262 auto select_rows = std::make_shared<phi::SelectedRows>();
263 out->set_impl(select_rows);
264 return select_rows.get();
265 }
266 return static_cast<phi::SelectedRows*>(out->impl().get());
267}
268
269phi::TensorBase* SetSparseKernelOutput(Tensor* out, TensorType type) {
270 if (!out) {

Callers 3

add_n_implFunction · 0.85
embedding_grad_implFunction · 0.85
copy_Method · 0.85

Calls 4

set_implMethod · 0.80
initializedMethod · 0.45
getMethod · 0.45
implMethod · 0.45

Tested by

no test coverage detected