Lines Matching refs:outputShape
38 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric() argument
41 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalGeneric()
43 std::vector<uint32_t> curIndex(outputShape.dimensions.size(), 0); in evalGeneric()
64 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8() argument
67 IndexedShapeWrapper outputShapeIndexed(outputShape); in evalQuant8()
69 std::vector<uint32_t> curIndex(outputShape.dimensions.size(), 0); in evalQuant8()
79 T aValue = requantize<T>(aData[aFlatIndex], aShape, outputShape); in evalQuant8()
80 T bValue = requantize<T>(bData[bFlatIndex], bShape, outputShape); in evalQuant8()
98 bool isMinimum, void* output, const Shape& outputShape) { in eval() argument
104 reinterpret_cast<_Float16*>(output), outputShape); in eval()
109 reinterpret_cast<float*>(output), outputShape); in eval()
114 reinterpret_cast<int32_t*>(output), outputShape); in eval()
119 reinterpret_cast<uint8_t*>(output), outputShape); in eval()
124 reinterpret_cast<int8_t*>(output), outputShape); in eval()