HomeSort by relevance Sort by last modified time
    Searched refs:MeanReducer (Results 1 - 13 of 13) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
reduction_ops_mean.cc 26 Eigen::internal::MeanReducer<type>>); \
32 Eigen::internal::MeanReducer<type>>);
45 Eigen::internal::MeanReducer<type>>); \
52 Eigen::internal::MeanReducer<type>>);
68 Eigen::internal::MeanReducer<type>>); \
75 Eigen::internal::MeanReducer<type>>);
reduction_ops.h 46 // MeanReducer is a special case, since it doesn't technically have an identity.
51 struct Identity<Eigen::internal::MeanReducer<T>> { \
52 static T identity(const Eigen::internal::MeanReducer<T>&) { \
reduction_ops_gpu_complex128.cu.cc 55 DEFINE_FOR_TYPE_AND_R(complex128, Eigen::internal::MeanReducer<complex128>);
reduction_ops_gpu_complex64.cu.cc 55 DEFINE_FOR_TYPE_AND_R(complex64, Eigen::internal::MeanReducer<complex64>);
reduction_ops_gpu_double.cu.cc 56 DEFINE_FOR_TYPE_AND_R(T, Eigen::internal::MeanReducer<T>); \
reduction_ops_gpu_float.cu.cc 56 DEFINE_FOR_TYPE_AND_R(T, Eigen::internal::MeanReducer<T>); \
reduction_ops_gpu_int.cu.cc 56 DEFINE_FOR_TYPE_AND_R(T, Eigen::internal::MeanReducer<T>); \
reduction_ops_half_mean_sum.cu.cc 56 DEFINE_FOR_TYPE_AND_R(T, Eigen::internal::MeanReducer<T>);
reduction_gpu_kernels.cu.h 843 struct ReduceFunctor<GPUDevice, Eigen::internal::MeanReducer<T>> {
847 const Eigen::internal::MeanReducer<T>& reducer) {
873 const Eigen::internal::MeanReducer<T>& reducer) {
879 struct ReduceFunctor<GPUDevice, Eigen::internal::MeanReducer<Eigen::half>> {
883 const Eigen::internal::MeanReducer<Eigen::half>& reducer) {
    [all...]
quantized_instance_norm.cc 351 reduction_indices, Eigen::internal::MeanReducer<float>());
358 .reduce(reduction_indices, Eigen::internal::MeanReducer<float>());
segment_reduction_ops.cc 311 "SegmentMean", Eigen::internal::MeanReducer<type>, type, index_type, 0); \
    [all...]
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorFunctors.h 141 template <typename T> struct MeanReducer
147 MeanReducer() : scalarCount_(0), packetCount_(0) { }
187 struct reducer_traits<MeanReducer<T>, Device> {
TensorBase.h 537 const TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const Dims, const Derived>
539 return TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const Dims, const Derived>(derived(), dims, internal::MeanReducer<CoeffReturnType>());
542 const TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>
545 return TensorReductionOp<internal::MeanReducer<CoeffReturnType>, const DimensionList<Index, NumDimensions>, const Derived>(derived(), in_dims, internal::MeanReducer<CoeffReturnType>());
    [all...]

Completed in 498 milliseconds