Searched refs:numReduceAxis (Results 1 – 1 of 1) sorted by relevance
623 int32_t numReduceAxis = axisSize; in meanPrepare() local636 --numReduceAxis; in meanPrepare()642 std::vector<uint32_t> outDims(numInputDims - numReduceAxis); in meanPrepare()