MCPcopy Create free account
hub / github.com/alibaba/MNN / _Size

Function _Size

express/NeuralNetWorkOp.cpp:972–976  ·  view source on GitHub ↗

Computes the size of the variable Args: input: A variable of type Halide_Type_Float or Halide_Type_Int Returns: A variable. The shape is (), and type is Halide_Type_Int */

Source from the content-addressed store, hash-verified

970A variable. The shape is (), and type is Halide_Type_Int
971*/
972VARP _Size(VARP input) {
973 std::unique_ptr<OpT> op(new OpT);
974 op->type = OpType_Size;
975 return (Variable::create(Expr::create(std::move(op), {input})));
976}
977
978/*Computes scaled exponential linear: scale * alpha * (exp(features) - 1) if < 0, scale * features otherwise.
979Args:

Callers 6

onExecuteMethod · 0.85
onExecuteMethod · 0.85
onExecuteMethod · 0.85
onExecuteMethod · 0.85
onGradMethod · 0.85
_runMethod · 0.85

Calls 1

createFunction · 0.50

Tested by 1

_runMethod · 0.68