Lines Matching refs:SimpleTensor
121 …c void unroll(ArithmeticOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
145 …c void unroll(ArithmeticOperation op, const SimpleTensor<T> &src1, const SimpleTensor<T> &src2, Si… in unroll()
154 SimpleTensor<T> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<T> &src1, const Sim… in arithmetic_operation()
166 SimpleTensor<uint8_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<uint8_t> &src… in arithmetic_operation()
170 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in arithmetic_operation()
171 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in arithmetic_operation()
172 …SimpleTensor<float> dst_tmp(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dst.data_typ… in arithmetic_operation()
196 SimpleTensor<int8_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int8_t> &src1,… in arithmetic_operation()
200 SimpleTensor<float> src1_tmp = convert_from_asymmetric(src1); in arithmetic_operation()
201 SimpleTensor<float> src2_tmp = convert_from_asymmetric(src2); in arithmetic_operation()
202 …SimpleTensor<float> dst_tmp(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dst.data_typ… in arithmetic_operation()
227 SimpleTensor<int16_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int16_t> &src… in arithmetic_operation()
231 SimpleTensor<float> src1_tmp = convert_from_symmetric<int16_t>(src1); in arithmetic_operation()
232 SimpleTensor<float> src2_tmp = convert_from_symmetric<int16_t>(src2); in arithmetic_operation()
233 …SimpleTensor<float> dst_tmp(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dst.data_typ… in arithmetic_operation()
257 …e SimpleTensor<int32_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int32_t> &…
259 …plate SimpleTensor<half> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<half> &sr…
260 …late SimpleTensor<float> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<float> &s…
263 SimpleTensor<T> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<T> &src1, const Sim… in arithmetic_operation()
267 SimpleTensor<T> dst(TensorShape::broadcast_shape(src1.shape(), src2.shape()), dst_data_type); in arithmetic_operation()
272 template SimpleTensor<int32_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int3…
274 template SimpleTensor<int16_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int1…
276 template SimpleTensor<int8_t> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<int8_…
277 template SimpleTensor<half> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<half> &…
278 template SimpleTensor<float> arithmetic_operation(ArithmeticOperation op, const SimpleTensor<float>…