Searched refs:SplitV (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | split_v.h | 31 class SplitV : public PrimitiveC {
34 SplitV() : PrimitiveC(kNameSplitV) { InitIOName({"input_x"}, {"output"}); } in SplitV() function
36 ~SplitV() = default;
37 MS_DECLARE_PARENT(SplitV, PrimitiveC);
42 using PrimSplitVPtr = std::shared_ptr<SplitV>;
|
| D | split_v.cc | 106 REGISTER_PRIMITIVE_EVAL_IMPL(SplitV, prim::kPrimSplitV, SplitVInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | split_npu.h | 50 hiai::op::SplitV *split_ = nullptr;
|
| D | split_npu.cc | 24 split_ = new (std::nothrow) hiai::op::SplitV(name_); in Init()
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 98 @bprop_getters.register(P.SplitV)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 30 … Shape, DynamicShape, Size, Slice, Split, SplitV, TransShape, ParallelConcat, Padding,
|
| D | array_ops.py | 6434 class SplitV(Primitive): class
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 2613 'block': P.SplitV(size_splits=[1, 2], split_dim=1, num_split=2),
|
| /third_party/mindspore/config/ |
| D | op_info.config | 473 {"op_name": "SplitV", "inputs": [{"index": 0, "name": "x", "need_compile": false, "param_type": "re…
|