Home
last modified time | relevance | path

Searched defs:split_sizes_ (Results 1 – 3 of 3) 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()
45 std::vector<int> split_sizes_{6, 6}; in TEST_F() local
67 std::vector<int> split_sizes_{2, 3, 3}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dglu_fp32.h60 int split_sizes_[kSplitNum] = {0}; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/
Dsplit_parameter.h36 int *split_sizes_; member