Home
last modified time | relevance | path

Searched defs:split_dim_ (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dsplit_tests.cc25 OpParameter *CreateParameter(int split_dim_, int num_split_, std::vector<int> split_sizes_) { in CreateParameter()
43 int split_dim_ = 3; in TEST_F() local
65 int split_dim_ = 0; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dsplit.h49 int split_dim_ = 0; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsplit.h74 int32_t split_dim_; variable
Dglu.h73 int32_t split_dim_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dsplit_parameter.h37 int split_dim_; member
51 int split_dim_; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/
Dsplit_npu.h52 hiai::op::Const *split_dim_ = nullptr; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dcholesky_trsm_solve_gpu_kernel.h291 int split_dim_; variable