Searched refs:kPrimJ (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | gradient_eliminate.cc | 138 if (IsPrimitiveCNode(node, prim::kPrimJ)) { in GetJPrim()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | manager.cc | 693 if (IsPrimitiveCNode(node, prim::kPrimJ)) { in OnEdgeAdded()
714 if (IsPrimitiveCNode(node, prim::kPrimJ)) { in OnEdgeRemoved()
1078 return prim->name() != prim::kPrimJ->name(); in SeekJ()
1094 if (IsPrimitiveCNode(iter.first, prim::kPrimJ)) { in SeekJ()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/ |
| D | dfunctor.cc | 374 if (IsPrimitiveCNode(fv, prim::kPrimJ)) { // Ignore if FV is a J CNode. in AttachFvDoutToTape()
851 } else if (IsPrimitive(cnode->inputs().at(0), prim::kPrimJ)) { in FindPrimalJPair()
|
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | ops_test.cc | 341 ASSERT_EQ(prim->name(), kPrimJ->name()); in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | prim_test.cc | 267 inputs.push_back(NewValueNode(prim::kPrimJ)); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
| D | ops_front_infer_function.cc | 673 REGISTER_PRIMITIVE_FRONT_EVAL_IMPL(J, prim::kPrimJ, InferImplJ, nullptr);
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 954 if (!IsPrimitiveCNode(expect_j_cnode, prim::kPrimJ)) { in FindSensNode()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | auto_monad.cc | 621 if (cnode->size() > 0 && IsPrimitiveCNode(cnode->input(0), prim::kPrimJ)) { in TraceTupleCNodeEffectInfo()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 697 inputs.push_back(NewValueNode(prim::kPrimJ)); in GenerateFuncGraph()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 626 inline const PrimitivePtr kPrimJ = std::make_shared<Primitive>("J", kSideEffectPropagate); variable
|