Lines Matching refs:shape1

77         const float* in1, const Shape& shape1, const float* in2, const Shape& shape2,
80 bool binaryOperationFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, in binaryOperationFloat16() argument
83 std::vector<float> in1_float32(getNumberOfElements(shape1)); in binaryOperationFloat16()
89 operationFloat32(in1_float32.data(), shape1, in2_float32.data(), shape2, activation, in binaryOperationFloat16()
96 bool addFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in addFloat32() argument
99 bool needBroadcast = !SameShape(shape1, shape2); in addFloat32()
104 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
113 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in addFloat32()
123 bool addFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in addFloat16() argument
126 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &addFloat32); in addFloat16()
130 bool addQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in addQuant8() argument
133 const bool needBroadcast = !SameShape(shape1, shape2); in addQuant8()
135 const int32_t input1_offset = -shape1.offset; in addQuant8()
139 const double twice_max_input_scale = 2 * std::max(shape1.scale, shape2.scale); in addQuant8()
140 const double real_input1_multiplier = shape1.scale / twice_max_input_scale; in addQuant8()
186 op_params, convertShapeToTflshape(shape1), in1, convertShapeToTflshape(shape2), in addQuant8()
190 tflite::reference_ops::BroadcastAdd4DSlow(op_params, convertShapeToTflshape(shape1), in addQuant8()
197 tflite::optimized_integer_ops::Add(op_params, convertShapeToTflshape(shape1), in1, in addQuant8()
202 tflite::optimized_ops::Add(op_params, convertShapeToTflshape(shape1), in1, in addQuant8()
235 bool mulFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in mulFloat32() argument
238 bool needBroadcast = !SameShape(shape1, shape2); in mulFloat32()
244 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), out, \ in mulFloat32()
254 tflite::optimized_ops::Mul(in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in mulFloat32()
262 bool mulFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in mulFloat16() argument
265 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &mulFloat32); in mulFloat16()
269 bool mulQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in mulQuant8() argument
272 const int32_t input1_offset = -shape1.offset; in mulQuant8()
275 const double input_product_scale = shape1.scale * shape2.scale; in mulQuant8()
303 tflite::reference_integer_ops::BroadcastMul4DSlow(op_params, convertShapeToTflshape(shape1), in mulQuant8()
308 tflite::reference_ops::BroadcastMul4DSlow(op_params, convertShapeToTflshape(shape1), in1, in mulQuant8()
316 bool subFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in subFloat32() argument
320 tflite::optimized_ops::Sub(in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in subFloat32()
333 bool subFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in subFloat16() argument
336 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &subFloat32); in subFloat16()
340 bool subQuant8(const T* in1, const Shape& shape1, const T* in2, const Shape& shape2, in subQuant8() argument
344 const int32_t input1_offset = -shape1.offset; in subQuant8()
348 const double twice_max_input_scale = 2 * std::max(shape1.scale, shape2.scale); in subQuant8()
349 const double real_input1_multiplier = shape1.scale / twice_max_input_scale; in subQuant8()
399 tflite::reference_integer_ops::BroadcastAdd4DSlow(op_params, convertShapeToTflshape(shape1), in subQuant8()
404 tflite::reference_ops::BroadcastAdd4DSlow(op_params, convertShapeToTflshape(shape1), in1, in subQuant8()
412 bool divFloat32(const float* in1, const Shape& shape1, const float* in2, const Shape& shape2, in divFloat32() argument
418 bool needBroadcast = !SameShape(shape1, shape2); in divFloat32()
422 in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in divFloat32()
426 tflite::optimized_ops::Div(in1, convertShapeToDims(shape1), in2, convertShapeToDims(shape2), in divFloat32()
433 bool divFloat16(const _Float16* in1, const Shape& shape1, const _Float16* in2, const Shape& shape2, in divFloat16() argument
436 return binaryOperationFloat16(in1, shape1, in2, shape2, activation, out, shapeOut, &divFloat32); in divFloat16()