• Home
  • Raw
  • Download

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()
242SimpleTensor<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()
248SimpleTensor<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()
270SimpleTensor<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()
276SimpleTensor<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()
298SimpleTensor<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()
304SimpleTensor<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()
326SimpleTensor<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()
332SimpleTensor<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…