MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / init_tile_repeat_state

Function init_tile_repeat_state

dnn/src/common/tile_repeat_helper.cpp:30–40  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

28
29template <typename T>
30void init_tile_repeat_state(
31 const T* src, T* dst, T* workspace0, T* /* workspace1 */, T*& current, T*& next,
32 size_t& state, size_t nr_reduces) {
33 current = const_cast<T*>(src);
34 if (nr_reduces == 1) {
35 next = dst;
36 } else {
37 next = workspace0;
38 }
39 state = 0;
40}
41
42template <typename T>
43void update_tile_repeat_state(

Callers 4

execMethod · 0.85
execMethod · 0.85
exec_internalMethod · 0.85
exec_internalMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected