Searched refs:sum_along_dim (Results 1 – 3 of 3) 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()
73 sum_along_dim = np.sum(out, axis=dim)75 np.ones(sum_along_dim.shape), sum_along_dim)
388 const Eigen::array<Eigen::DenseIndex, 1> sum_along_dim(0); in TEST_F() local411 y_square_sum = y.matrix<float>().square().sum(sum_along_dim); in TEST_F()414 error_square_sum = (x_matrix - y_normalized).square().sum(sum_along_dim); in TEST_F()