Searched refs:EuclideanNormReducer (Results 1 – 9 of 9) sorted by relevance
26 functor::EuclideanNormReducer<type>>); \32 functor::EuclideanNormReducer<type>>);45 functor::EuclideanNormReducer<type>>); \52 functor::EuclideanNormReducer<type>>);
42 struct EuclideanNormReducer { struct47 struct ReducerTraits<EuclideanNormReducer<Scalar>> { argument130 functor::EuclideanNormReducer<Scalar>> {133 const functor::EuclideanNormReducer<Scalar>& reducer) {144 functor::EuclideanNormReducer<bfloat16>> {147 const functor::EuclideanNormReducer<bfloat16>& reducer) {
56 DEFINE_FOR_TYPE_AND_R(T, functor::EuclideanNormReducer<T>); \
56 DEFINE_FOR_TYPE_AND_R(complex64, functor::EuclideanNormReducer<complex64>);
56 DEFINE_FOR_TYPE_AND_R(complex128, functor::EuclideanNormReducer<complex128>);
59 DEFINE_FOR_TYPE_AND_R(T, functor::EuclideanNormReducer<T>); \
1140 struct ReduceFunctor<GPUDevice, functor::EuclideanNormReducer<T>> {1144 const functor::EuclideanNormReducer<T>& reducer) {1158 const functor::EuclideanNormReducer<T>& reducer) {