Searched refs:kPrimAbs (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | abs.h | 33 Abs() : PrimitiveC(prim::kPrimAbs->name()) { InitIOName({"input_x"}, {"output"}); } in Abs()
|
| D | abs.cc | 134 REGISTER_PRIMITIVE_EVAL_IMPL(Abs, prim::kPrimAbs, AbsInfer, AbsInferValue, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | eltwise_cpu_kernel.cc | 41 {prim::kPrimAbs->name(), DescParam{dnnl::algorithm::eltwise_abs}}, in GetForwardEltwiseDesc()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_cluster.cc | 35 {kAllTarget, OpLevel_0, prim::kPrimAbs}, in GetClusterableOpList()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 504 inline const PrimitivePtr kPrimAbs = std::make_shared<Primitive>(kAbs); variable
|