Searched refs:GetSparseElement (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 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()
867 GetSparseElement<half>(sparse_element_number, shape_index))); in GetSparseElementAsString()
[all …]
|
| D | literal.h | 149 NativeT GetSparseElement(int64 sparse_element_number,
950 NativeT LiteralBase::GetSparseElement(int64 sparse_element_number, in GetSparseElement() function
|