OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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