Searched refs:sum_along_dim (Results 1 – 1 of 1) sorted by relevance
321 Eigen::array<TensorIndex, 1> sum_along_dim; in rowReduction() local322 sum_along_dim[0] = 0; in rowReduction()326 Eigen::IndexList<Eigen::type2index<0>> sum_along_dim; in rowReduction() local331 C.device(device_) = B.sum(sum_along_dim); in rowReduction()351 Eigen::array<TensorIndex, 1> sum_along_dim; in colReduction() local352 sum_along_dim[0] = 1; in colReduction()356 Eigen::IndexList<Eigen::type2index<1>> sum_along_dim; in colReduction() local361 C.device(device_) = B.sum(sum_along_dim); in colReduction()