Home
last modified time | relevance | path

Searched defs:scale_prim (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/onnx/
Donnx_einsum_adjust.cc124 auto scale_prim = scale_node.GetPrim(); in Adjust() local
143 auto scale_prim = matmul_node.GetPrim(); in Adjust() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dscale_activation_fusion.cc63 auto scale_prim = ops::GetOperator<ops::ScaleFusion>(scale_cnode->input(FIRST_INPUT)); in Process() local
Dscale_scale_fusion.cc56 auto scale_prim = ops::GetOperator<ops::ScaleFusion>(scale_cnode->input(FIRST_INPUT)); in CheckScaleNode() local
294 auto scale_prim = ops::GetOperator<ops::ScaleFusion>(up_scale_cnode->input(FIRST_INPUT)); in Process() local
Dscale_base_fusion.cc122 auto scale_prim = ops::GetOperator<ops::ScaleFusion>(scale_cnode->input(0)); in CheckCurrCnodeProper() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/mapper/
Dscale_mapper.cc78 auto scale_prim = api::utils::cast<api::SharedPtr<ops::Scale>>(prim); in Map() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/
Dscale_npu.cc28 auto scale_prim = primitive->value_as_ScaleFusion(); in IsSupport() local
57 auto scale_prim = primitive->value_as_ScaleFusion(); in Init() local