Lines Matching refs:outputShape
49 float* outputData, const Shape& outputShape) { in l2normFloat32Impl() argument
77 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8Impl() argument
109 int8_t* outputData, const Shape& outputShape) { in l2normQuant8SignedImpl() argument
140 const Shape& outputShape) { in l2normFloat32() argument
148 convertShapeToTflshape(outputShape), outputData); in l2normFloat32()
151 return l2normFloat32Impl(inputData, inputShape, axis, outputData, outputShape); in l2normFloat32()
156 _Float16* outputData, const Shape& outputShape) { in l2normFloat16() argument
160 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in l2normFloat16()
162 l2normFloat32(inputDataFloat32.data(), inputShape, axis, outputDataFloat32.data(), outputShape); in l2normFloat16()
169 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8() argument
177 convertShapeToTflshape(outputShape), outputData); in l2normQuant8()
180 return l2normQuant8Impl(inputData, inputShape, axis, outputData, outputShape); in l2normQuant8()
185 int8_t* outputData, const Shape& outputShape) { in l2normQuant8Signed() argument
197 return l2normQuant8SignedImpl(inputData, inputShape, axis, outputData, outputShape); in l2normQuant8Signed()