Searched refs:kPrimSquare (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | adam_apply_one_with_decay_rule.cc | 52 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
72 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
92 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
112 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
132 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
152 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
177 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
202 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
227 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
252 VectorRef square0({prim::kPrimSquare, input0_}); in DefinePattern()
|
| D | adam_apply_one_fusion.cc | 27 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
42 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
57 …VectorRef({prim::kPrimMul, VectorRef({prim::kPrimSquare, input_vars_[kIndex0]}), mul_x_input_vars_… in DefinePattern()
72 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
87 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
102 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
124 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
146 …VectorRef({prim::kPrimMul, VectorRef({prim::kPrimSquare, input_vars_[kIndex0]}), mul_x_input_vars_… in DefinePattern()
168 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
190 …VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex3], VectorRef({prim::kPrimSquare, input_vars_[k… in DefinePattern()
|
| D | lamb_next_right_rule.cc | 54 VectorRef mul3 = VectorRef({prim::kPrimMul, mul3_x_, VectorRef({prim::kPrimSquare, input0_})}); in DefinePattern()
|
| D | square_sum_fusion.cc | 96 return VectorRef({prim::kPrimReduceSum, VectorRef({prim::kPrimSquare, X})}); in DefinePattern()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | square.h | 32 Square() : PrimitiveC(prim::kPrimSquare->name()) { InitIOName({"input_x"}, {"output"}); } in Square()
|
| D | square.cc | 131 REGISTER_PRIMITIVE_EVAL_IMPL(Square, prim::kPrimSquare, SquareInfer, SquareInferValue, true);
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | value_based_eliminate.cc | 77 if (IsPrimitiveCNode(node, prim::kPrimSquare) || IsPrimitiveCNode(node, prim::kPrimSqrt)) { in IsCNodePositive()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | adam_fusion.cc | 99 … VectorRef({prim::kPrimMul, one_sub_beta2_, VectorRef({prim::kPrimSquare, gradient_})})}); in DefinePattern()
|
| D | adam_weight_decay_fusion.cc | 99 … VectorRef({prim::kPrimMul, one_sub_beta2_, VectorRef({prim::kPrimSquare, gradient_})})}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 59 {kAllTarget, OpLevel_0, prim::kPrimSquare}, in GetExpandOps()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_self_cpu_kernel.cc | 311 {prim::kPrimSquare->name(), Square<T>}, in LaunchKernel()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 490 inline const PrimitivePtr kPrimSquare = std::make_shared<Primitive>(kSquare); variable
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1620 if (node->IsApply(prim::kPrimSquare)) { in ExportCNode()
|