Home
last modified time | relevance | path

Searched refs:sparse_element_number (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dsparse_index_array.cc52 int64 sparse_element_number) const { in At()
54 CHECK_GE(sparse_element_number, 0); in At()
55 CHECK_LE(rank_ * sparse_element_number + rank_, indices_.size()); in At()
57 indices_.data() + rank_ * sparse_element_number, rank_); in At()
60 absl::Span<int64> SparseIndexArray::At(int64 sparse_element_number) { in At() argument
62 CHECK_GE(sparse_element_number, 0); in At()
63 CHECK_LE(rank_ * sparse_element_number + rank_, indices_.size()); in At()
64 return absl::Span<int64>(indices_.data() + rank_ * sparse_element_number, in At()
Dliteral.cc834 int64 sparse_element_number, const ShapeIndex& shape_index) const { in GetSparseElementAsString() argument
839 return GetSparseElement<bool>(sparse_element_number, shape_index) in GetSparseElementAsString()
843 return StrCat(GetSparseElement<int8>(sparse_element_number, shape_index)); in GetSparseElementAsString()
846 GetSparseElement<int16>(sparse_element_number, shape_index)); in GetSparseElementAsString()
849 GetSparseElement<int32>(sparse_element_number, shape_index)); in GetSparseElementAsString()
852 GetSparseElement<int64>(sparse_element_number, shape_index)); in GetSparseElementAsString()
855 GetSparseElement<uint8>(sparse_element_number, shape_index)); in GetSparseElementAsString()
858 GetSparseElement<uint16>(sparse_element_number, shape_index)); in GetSparseElementAsString()
861 GetSparseElement<uint32>(sparse_element_number, shape_index)); in GetSparseElementAsString()
864 GetSparseElement<uint64>(sparse_element_number, shape_index)); in GetSparseElementAsString()
[all …]
Dsparse_index_array.h76 absl::Span<const int64> At(int64 sparse_element_number) const;
77 absl::Span<int64> At(int64 sparse_element_number);
Dliteral.h131 string GetSparseElementAsString(int64 sparse_element_number,
142 int64 sparse_element_number, const ShapeIndex& shape_index = {}) const;
149 NativeT GetSparseElement(int64 sparse_element_number,
950 NativeT LiteralBase::GetSparseElement(int64 sparse_element_number, in GetSparseElement() argument
954 return data<NativeT>(shape_index)[sparse_element_number]; in GetSparseElement()