Searched refs:kPrimReciprocal (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | reciprocal.h | 33 Reciprocal() : PrimitiveC(prim::kPrimReciprocal->name()) { InitIOName({"x"}, {"y"}); } in Reciprocal()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 220 {prim::kPrimReciprocal, {ops::ReciprocalInfer, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | pattern_matcher.h | 736 if (calcu_type == prim::kPrimReciprocal) {
758 if (data2[i] == 0 && calcu_type == prim::kPrimReciprocal) {
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_self_cpu_kernel.cc | 329 {prim::kPrimReciprocal->name(), Reciprocal<T>}, in LaunchKernel()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_cluster.cc | 48 {kAllTarget, OpLevel_0, prim::kPrimReciprocal}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 502 inline const PrimitivePtr kPrimReciprocal = std::make_shared<Primitive>(kReciprocal); variable
|