Home
last modified time | relevance | path

Searched defs:sum_along_dim (Results 1 – 2 of 2) sorted by relevance

/external/eigen/bench/tensors/
Dtensor_benchmarks.h321 Eigen::array<TensorIndex, 1> sum_along_dim; in rowReduction() local
326 Eigen::IndexList<Eigen::type2index<0>> sum_along_dim; in rowReduction() local
351 Eigen::array<TensorIndex, 1> sum_along_dim; in colReduction() local
356 Eigen::IndexList<Eigen::type2index<1>> sum_along_dim; in colReduction() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc388 const Eigen::array<Eigen::DenseIndex, 1> sum_along_dim(0); in TEST_F() local