• Home
  • Raw
  • Download

Lines Matching refs:ReductionOperation

41 OT reduce_operation(const T *ptr, int reduce_elements, ReductionOperation op, int stride, RoundingP…  in reduce_operation()
47 case ReductionOperation::PROD: in reduce_operation()
52 case ReductionOperation::MIN: in reduce_operation()
53 case ReductionOperation::MAX: in reduce_operation()
73 case ReductionOperation::MIN: in reduce_operation()
79 case ReductionOperation::MAX: in reduce_operation()
85 case ReductionOperation::SUM_SQUARE: in reduce_operation()
88 case ReductionOperation::MEAN_SUM: in reduce_operation()
89 case ReductionOperation::SUM: in reduce_operation()
92 case ReductionOperation::PROD: in reduce_operation()
99 if(op == ReductionOperation::MEAN_SUM && reduce_elements > 0) in reduce_operation()
119 case ReductionOperation::MIN: in reduce_operation()
125 case ReductionOperation::MAX: in reduce_operation()
131 case ReductionOperation::SUM_SQUARE: in reduce_operation()
134 case ReductionOperation::MEAN_SUM: in reduce_operation()
135 case ReductionOperation::SUM: in reduce_operation()
138 case ReductionOperation::PROD: in reduce_operation()
145 if(op == ReductionOperation::MEAN_SUM && reduce_elements > 0) in reduce_operation()
154 OT reduce_operation_arg_min_max(const T *ptr, int reduce_elements, ReductionOperation op, int strid… in reduce_operation_arg_min_max()
162 case ReductionOperation::ARG_IDX_MIN: in reduce_operation_arg_min_max()
168 case ReductionOperation::ARG_IDX_MAX: in reduce_operation_arg_min_max()
184 …leTensor<T> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, Rounding… in compute_reduction_operation()
187 …const bool is_arg_min_max = (op == ReductionOperation::ARG_IDX_MIN || op == ReductionOpe… in compute_reduction_operation()
278 …leTensor<T> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, Quantiza… in reduction_operation()
285 …or<uint8_t> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, Quantiza… in reduction_operation()
290 … if(op == ReductionOperation::MEAN_SUM && src.quantization_info() == quantization_info_output) in reduction_operation()
308 …sor<int8_t> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op, Quantiza… in reduction_operation()
313 … if(op == ReductionOperation::MEAN_SUM && src.quantization_info() == quantization_info_output) in reduction_operation()
330 …t SimpleTensor<float> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op,
332 …st SimpleTensor<half> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op,
335 …t SimpleTensor<float> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op,
337 …SimpleTensor<int32_t> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op,
339 …st SimpleTensor<half> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op,
341 …SimpleTensor<uint8_t> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op,
343 … SimpleTensor<int8_t> &src, const TensorShape &dst_shape, unsigned int axis, ReductionOperation op,