Searched refs:ReducerFirst (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | reduce.h | 326 template <typename T, typename U, typename ReducerFirst, typename ReducerNext>
330 const ReducerFirst& reducer_first, in ReduceImpl()
385 template <typename In, typename Out, typename ReducerFirst,
390 const ReducerFirst& reducer_first, in Reduce()
596 struct ReducerFirst { struct
597 explicit ReducerFirst(int input_zero_point_arg) in ReducerFirst() argument
639 if (!Reduce<T, int32_t, ReducerFirst<T>, ReducerNext<T>>( in QuantizedReduceProd()
641 num_resolved_axis, temp_prod, ReducerFirst<T>(input_zero_point), in QuantizedReduceProd()
|