Searched defs:reduce_dim (Results 1 – 10 of 10) sorted by relevance
76 Eigen::array<int, 1> reduce_dim({0}); in Compute() local159 Eigen::array<int, 1> reduce_dim({0}); in Compute() local
68 for (int32_t reduce_dim : reduced_indices) { in GetReductionIterSize() local
492 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; in HandleReduce() local
48 def _joined_array(num_dims, reduce_dim): argument
93 int64_t reduce_dim = n_dims - 1; in GetMatrixDiagonal() local
1666 auto reduce_dim = in matchAndRewrite() local
824 const int32_t reduce_dim, in Softmax()
823 const int32_t reduce_dim, in Softmax()
1048 int reduce_dim = in TEST_F() local1064 int reduce_dim = in TEST_F() local