Home
last modified time | relevance | path

Searched refs:mul_prim (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Dmul_populate_v0.cc27 auto mul_prim = primitive->value_as_Mul(); in PopulateMulParameter() local
28 if (mul_prim == nullptr) { in PopulateMulParameter()
38 param->activation_type_ = mul_prim->activationType(); in PopulateMulParameter()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Darithmetic_simplify.cc124 auto mul_prim = mul_cnode_->cast<CNodePtr>()->input(0); in operator ()() local
165 AnfNodePtr mul = NewCNode({mul_prim, all_reduce, y.GetNode(node)}, fg); in operator ()()
Dspecial_op_eliminate.h672 auto mul_prim = prim::GetPythonOps("tensor_mul", "mindspore.ops.functional"); in operator() local
673 MS_EXCEPTION_IF_NULL(mul_prim); in operator()
674 mul_inputs.push_back(NewValueNode(mul_prim)); in operator()