Searched refs:kPrimEqual (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | equal.h | 34 Equal() : PrimitiveC(prim::kPrimEqual->name()) { InitIOName({"x", "y"}, {"output"}); } in Equal()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | arithmetic_logic_cpu_kernel.cc | 178 {prim::kPrimEqual->name(), &ArithmeticLogicCPUKernel<T>::Equal}}; in InitComputeFunc()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 217 {prim::kPrimEqual, {ops::EqualInfer, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_cluster.cc | 38 {kAllTarget, OpLevel_0, prim::kPrimEqual}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 142 inline const PrimitivePtr kPrimEqual = std::make_shared<Primitive>(kEqual); variable
|