Home | History | Annotate | Download | only in xla

Lines Matching defs:rank

32 // sparse index is an array of `rank` int64 value that gives the location of a
34 // checked (except for the rank). To avoid confusion, we refer to the position
45 // indices, with an initial contents obtained from the given array. The rank
64 SparseIndexArray(int64 max_indices, int64 rank,
66 SparseIndexArray(int64 max_indices, int64 rank,
94 int64 rank() const { return rank_; }
142 tensorflow::gtl::InlinedVector<int64, 8> saved_index(rank());
153 std::copy_n(At(i).begin(), rank(), saved_index.begin());
158 std::copy_n(saved_index.begin(), rank(), At(j).begin());
164 std::copy_n(At(sort_order[j]).begin(), rank(), At(j).begin());