Searched refs:group_dims (Results 1 – 2 of 2) sorted by relevance
148 std::vector<int64> group_dims(truth_st_shape.dims() - 1); in Compute() local149 std::iota(group_dims.begin(), group_dims.end(), 0); in Compute()152 for (int d = 0; d < static_cast<int>(group_dims.size()); ++d) { in Compute()168 auto hypothesis_grouper = hypothesis.group(group_dims); in Compute()169 auto truth_grouper = truth.group(group_dims); in Compute()
80 GroupIterable(Tensor ix, Tensor vals, int dims, const VarDimArray& group_dims) in GroupIterable() argument85 group_dims_(group_dims.begin(), group_dims.end()) {} in GroupIterable()