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

Method comp_node

imperative/src/impl/transformations/eval.cpp:13–19  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

11}
12
13CompNodeValue::ref_t InterpreterValue::comp_node() const {
14 if (!m_comp_node) {
15 m_comp_node = CompNodeValue::make(
16 handle()->channel()->get_device(handle()->handle()));
17 }
18 return m_comp_node;
19}
20
21ShapeValue::ref_t InterpreterValue::shape() const {
22 if (!m_shape) {

Callers 15

TESTFunction · 0.45
check_rng_basicFunction · 0.45
TESTFunction · 0.45
freeMethod · 0.45
InputPlaceholderFunction · 0.45
makeFunction · 0.45
make_backward_graphMethod · 0.45
as_tensorMethod · 0.45
ProfilerPluginMethod · 0.45

Calls 4

get_deviceMethod · 0.80
makeFunction · 0.50
channelMethod · 0.45
handleMethod · 0.45

Tested by 6

TESTFunction · 0.36
check_rng_basicFunction · 0.36
TESTFunction · 0.36