Home
last modified time | relevance | path

Searched refs:splitPrepare (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/include/
DOperationsUtils.h333 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, std::vector<Shape>* output);
/frameworks/ml/nn/common/
DOperationsUtils.cpp691 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, in splitPrepare() function
DCpuExecutor.cpp1442 success = splitPrepare(input.shape(), axis, numOutputs, &outputShapes); in executeOperation()