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

Method check

src/plugin/impl/num_range_checker.cpp:26–37  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

24}
25
26bool NumRangeChecker::Checker::check(VarNode* var) {
27 auto&& val = var->dev_tensor();
28 if (val.layout().is_contiguous()) {
29 *m_inp = var->dev_tensor();
30 } else {
31 *m_inp = {};
32 m_inp->copy_from(val);
33 }
34 m_func->execute();
35 mgb_assert(m_out->shape().is_scalar());
36 return m_out->ptr<float>()[0] >= 0.5;
37}
38
39NumRangeChecker::NumRangeChecker(cg::ComputingGraph* graph, float range)
40 : PluginBase(graph), m_range{range} {

Callers 1

on_var_computedMethod · 0.45

Calls 7

dev_tensorMethod · 0.45
is_contiguousMethod · 0.45
layoutMethod · 0.45
copy_fromMethod · 0.45
executeMethod · 0.45
is_scalarMethod · 0.45
shapeMethod · 0.45

Tested by

no test coverage detected