Lines Matching refs:SimpleTensor
182 …static void unroll(const SimpleTensor<T1> &src1, const SimpleTensor<T2> &src2, SimpleTensor<T3> &d… in unroll()
207 …static void unroll(const SimpleTensor<T1> &src1, const SimpleTensor<T2> &src2, SimpleTensor<T3> &d… in unroll()
217 SimpleTensor<T3> pixel_wise_multiplication(const SimpleTensor<T1> &src1, const SimpleTensor<T2> &sr… in pixel_wise_multiplication()
222 SimpleTensor<T3> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dt_out); in pixel_wise_multiplication()
239 SimpleTensor<uint8_t> pixel_wise_multiplication(const SimpleTensor<uint8_t> &src1, const SimpleTens… in pixel_wise_multiplication()
242 …SimpleTensor<uint8_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dt_out, 1, qou… in pixel_wise_multiplication()
246 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in pixel_wise_multiplication()
247 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in pixel_wise_multiplication()
248 …SimpleTensor<float> dst_tmp = pixel_wise_multiplication<float, float, float>(src1_tmp, src2_tmp, … in pixel_wise_multiplication()
267 SimpleTensor<int16_t> pixel_wise_multiplication(const SimpleTensor<uint8_t> &src1, const SimpleTens… in pixel_wise_multiplication()
270 …SimpleTensor<int16_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dt_out, 1, qou… in pixel_wise_multiplication()
274 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in pixel_wise_multiplication()
275 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in pixel_wise_multiplication()
276 …SimpleTensor<float> dst_tmp = pixel_wise_multiplication<float, float, float>(src1_tmp, src2_tmp, … in pixel_wise_multiplication()
295 SimpleTensor<int8_t> pixel_wise_multiplication(const SimpleTensor<int8_t> &src1, const SimpleTensor… in pixel_wise_multiplication()
298 …SimpleTensor<int8_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dt_out, 1, qout… in pixel_wise_multiplication()
302 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in pixel_wise_multiplication()
303 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in pixel_wise_multiplication()
304 …SimpleTensor<float> dst_tmp = pixel_wise_multiplication<float, float, float>(src1_tmp, src2_tmp, … in pixel_wise_multiplication()
323 SimpleTensor<int16_t> pixel_wise_multiplication(const SimpleTensor<int16_t> &src1, const SimpleTens… in pixel_wise_multiplication()
326 …SimpleTensor<int16_t> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dt_out, 1, qou… in pixel_wise_multiplication()
330 SimpleTensor<float> src1_tmp = convert_from_symmetric<int16_t>(src1); in pixel_wise_multiplication()
331 SimpleTensor<float> src2_tmp = convert_from_symmetric<int16_t>(src2); in pixel_wise_multiplication()
332 …SimpleTensor<float> dst_tmp = pixel_wise_multiplication<float, float, float>(src1_tmp, src2_tmp, … in pixel_wise_multiplication()
351 template SimpleTensor<int16_t> pixel_wise_multiplication(const SimpleTensor<uint8_t> &src1, const S…
352 template SimpleTensor<int32_t> pixel_wise_multiplication(const SimpleTensor<int16_t> &src1, const S…
353 template SimpleTensor<int32_t> pixel_wise_multiplication(const SimpleTensor<int32_t> &src1, const S…
354 template SimpleTensor<float> pixel_wise_multiplication(const SimpleTensor<float> &src1, const Simpl…
355 template SimpleTensor<half_float::half> pixel_wise_multiplication(const SimpleTensor<half_float::ha…