Searched defs:reduce_dim (Results 1 – 12 of 12) sorted by relevance
78 Eigen::array<int, 1> reduce_dim({0}); in Compute() local161 Eigen::array<int, 1> reduce_dim({0}); in Compute() local
69 for (int32_t reduce_dim : reduced_indices) { in GetReductionIterSize() local
492 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; in HandleReduce() local
43 def _joined_array(num_dims, reduce_dim): argument
68 std::string reduce_dim; in ExpandOp() local
93 int64_t reduce_dim = n_dims - 1; in GetMatrixDiagonal() local
1658 auto reduce_dim = in matchAndRewrite() local
1507 auto reduce_dim = (*dimensions.value_begin<APInt>()).getSExtValue(); in MatchIotaConst() local
823 const int32_t reduce_dim, in Softmax()
824 const int32_t reduce_dim, in Softmax()
1684 int reduce_dim = in TEST_F() local1701 int reduce_dim = in TEST_F() local