Searched refs:kPrimScalarToArray (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | resource.cc | 72 … {"__ms_to_array__", prim::kPrimScalarToArray}, // P.scalar_to_array in GetMethodMap()
94 … {"__ms_to_array__", prim::kPrimScalarToArray}, // P.scalar_to_array, in GetMethodMap()
116 … {"__ms_to_array__", prim::kPrimScalarToArray}, // P.scalar_to_array, in GetMethodMap()
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ad/ |
| D | ad_test.cc | 165 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimScalarToArray), resourcePtr); in TEST_F()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 109 {prim::kPrimScalarToArray, {InferImplScalarToArray, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | ops_test.cc | 258 ASSERT_EQ(prim->name(), kPrimScalarToArray->name()); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | composite.cc | 329 …ack(std::make_pair(func_graph->NewCNodeInOrder({NewValueNode(prim::kPrimScalarToArray), item.first… in Harmonize()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 167 inline const PrimitivePtr kPrimScalarToArray = std::make_shared<Primitive>("scalar_to_array"); variable
|