Searched refs:sum_along_dim (Results 1 - 3 of 3) sorted by relevance

/external/eigen/bench/tensors/
H A Dtensor_benchmarks.h321 Eigen::array<TensorIndex, 1> sum_along_dim; local
322 sum_along_dim[0] = 0;
326 Eigen::IndexList<Eigen::type2index<0>> sum_along_dim; local
331 C.device(device_) = B.sum(sum_along_dim);
351 Eigen::array<TensorIndex, 1> sum_along_dim; local
352 sum_along_dim[0] = 1;
356 Eigen::IndexList<Eigen::type2index<1>> sum_along_dim; local
361 C.device(device_) = B.sum(sum_along_dim);
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dsoftmax_op_test.py65 sum_along_dim = np.sum(out, axis=dim)
67 np.ones(sum_along_dim.shape), sum_along_dim)
/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dmaster_test.cc388 const Eigen::array<Eigen::DenseIndex, 1> sum_along_dim(0);
411 y_square_sum = y.matrix<float>().square().sum(sum_along_dim);
414 error_square_sum = (x_matrix - y_normalized).square().sum(sum_along_dim);

Completed in 1161 milliseconds