Searched refs:splitPrepare (Results 1 – 3 of 3) sorted by relevance
317 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, std::vector<Shape>* output);
689 bool splitPrepare(const Shape& input, int32_t axis, int32_t numOutputs, in splitPrepare() function
1386 success = splitPrepare(input.shape(), axis, numOutputs, &outputShapes); in executeOperation()