Lines Matching refs:outputShape
36 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric() argument
39 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()
41 std::vector<uint32_t> curIndex(outputShape.dimensions.size(), 0); in evalGeneric()
62 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8() argument
65 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalQuant8()
67 std::vector<uint32_t> curIndex(outputShape.dimensions.size(), 0); in evalQuant8()
77 T aValue = requantize<T>(aData[aFlatIndex], aShape, outputShape); in evalQuant8()
78 T bValue = requantize<T>(bData[bFlatIndex], bShape, outputShape); in evalQuant8()
96 bool isMinimum, void* output, const Shape& outputShape) { in eval() argument
102 reinterpret_cast<_Float16*>(output), outputShape); in eval()
107 reinterpret_cast<float*>(output), outputShape); in eval()
112 reinterpret_cast<int32_t*>(output), outputShape); in eval()
117 reinterpret_cast<uint8_t*>(output), outputShape); in eval()
122 reinterpret_cast<int8_t*>(output), outputShape); in eval()