Lines Matching refs:shapeOut

78         int32_t activation, float* out, const Shape& shapeOut)>;
82 const Shape& shapeOut, binaryFunctionFloat32 operationFloat32) { in binaryOperationFloat16() argument
87 std::vector<float> out_float32(getNumberOfElements(shapeOut)); in binaryOperationFloat16()
90 out_float32.data(), shapeOut); in binaryOperationFloat16()
97 int32_t activation, float* out, const Shape& shapeOut) { in addFloat32() argument
105 convertShapeToDims(shapeOut)) in addFloat32()
114 convertShapeToDims(shapeOut)) in addFloat32()
124 int32_t activation, _Float16* out, const Shape& shapeOut) { in addFloat16() argument
126 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &addFloat32); in addFloat16()
131 int32_t activation, T* out, const Shape& shapeOut) { in addQuant8() argument
137 const int32_t output_offset = shapeOut.offset; in addQuant8()
143 twice_max_input_scale / ((1 << left_shift) * shapeOut.scale); in addQuant8()
162 CalculateActivationRangeInt8(activation, shapeOut, &output_activation_min, in addQuant8()
165 CalculateActivationRangeUint8(activation, shapeOut, &output_activation_min, in addQuant8()
187 in2, convertShapeToTflshape(shapeOut), out); in addQuant8()
192 convertShapeToTflshape(shapeOut), out); in addQuant8()
199 convertShapeToTflshape(shapeOut), out); in addQuant8()
204 convertShapeToTflshape(shapeOut), out); in addQuant8()
236 int32_t activation, float* out, const Shape& shapeOut) { in mulFloat32() argument
245 convertShapeToDims(shapeOut)) in mulFloat32()
256 convertShapeToDims(shapeOut)); in mulFloat32()
263 int32_t activation, _Float16* out, const Shape& shapeOut) { in mulFloat16() argument
265 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &mulFloat32); in mulFloat16()
270 int32_t activation, T* out, const Shape& shapeOut) { in mulQuant8() argument
274 const int32_t output_offset = shapeOut.offset; in mulQuant8()
276 const double real_multiplier = input_product_scale / shapeOut.scale; in mulQuant8()
286 CalculateActivationRangeInt8(activation, shapeOut, &output_activation_min, in mulQuant8()
289 CalculateActivationRangeUint8(activation, shapeOut, &output_activation_min, in mulQuant8()
305 convertShapeToTflshape(shapeOut), out); in mulQuant8()
310 convertShapeToTflshape(shapeOut), out); in mulQuant8()
317 int32_t activation, float* out, const Shape& shapeOut) { in subFloat32() argument
321 out, convertShapeToDims(shapeOut)); in subFloat32()
326 uint32_t numOutputElements = getNumberOfElements(shapeOut); in subFloat32()
334 int32_t activation, _Float16* out, const Shape& shapeOut) { in subFloat16() argument
336 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &subFloat32); in subFloat16()
341 int32_t activation, T* out, const Shape& shapeOut) { in subQuant8() argument
346 const int32_t output_offset = shapeOut.offset; in subQuant8()
352 twice_max_input_scale / ((1 << left_shift) * shapeOut.scale); in subQuant8()
374 CalculateActivationRangeInt8(activation, shapeOut, &output_activation_min, in subQuant8()
377 CalculateActivationRangeUint8(activation, shapeOut, &output_activation_min, in subQuant8()
401 convertShapeToTflshape(shapeOut), out); in subQuant8()
406 convertShapeToTflshape(shapeOut), out); in subQuant8()
413 int32_t activation, float* out, const Shape& shapeOut) { in divFloat32() argument
423 output_activation_min, output_activation_max, out, convertShapeToDims(shapeOut)); in divFloat32()
428 convertShapeToDims(shapeOut)); in divFloat32()
434 int32_t activation, _Float16* out, const Shape& shapeOut) { in divFloat16() argument
436 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &divFloat32); in divFloat16()