Searched defs:multi_index (Results 1 – 12 of 12) sorted by relevance
50 absl::Span<const int64> multi_index) { in CompareFloatsBitwiseEqual()61 absl::Span<const int64> multi_index) { in CompareEqual()69 absl::Span<const int64> multi_index) { in CompareEqual()74 absl::Span<const int64> multi_index) { in CompareEqual()79 absl::Span<const int64> multi_index) { in CompareEqual()84 absl::Span<const int64> multi_index) { in CompareEqual()89 absl::Span<const int64> multi_index) { in CompareEqual()95 absl::Span<const int64> multi_index) { in CompareEqual()102 absl::Span<const int64> multi_index) { in MakeBitwiseErrorStatus()117 absl::Span<const int64> multi_index) { in MakeErrorStatus()[all …]
29 const Shape& shape, absl::Span<const int64> multi_index) { in MultidimensionalIndexToLinearIndex()104 std::vector<int64> multi_index(shape.dimensions_size()); in LinearIndexToMultidimensionalIndex() local
895 NativeT LiteralBase::Piece::Get(absl::Span<const int64> multi_index) const { in Get()902 void LiteralBase::Piece::Set(absl::Span<const int64> multi_index, in Set()921 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index, in Get()927 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index) const { in Get()932 inline void MutableLiteralBase::Set(absl::Span<const int64> multi_index, in Set()939 inline void MutableLiteralBase::Set(absl::Span<const int64> multi_index, in Set()959 absl::Span<const int64> multi_index, NativeT value, in AppendSparseElement()
40 std::vector<int64> multi_index = in TEST() local133 std::vector<int64> multi_index = in TEST() local
458 absl::Span<const int64> multi_index) { in MultiIndexAsString()
216 absl::Span<const int64> multi_index) { in CopySliceFromInternal()788 string LiteralBase::GetAsString(absl::Span<const int64> multi_index, in GetAsString()936 Status MutableLiteralBase::SetIntegralAsS64(absl::Span<const int64> multi_index, in SetIntegralAsS64()1485 std::vector<int64> multi_index; in EqualElements() local
244 std::vector<llvm::Value*> multi_index(shape.dimensions_size()); in AddLoopsForShapeOnDimensions() local272 std::vector<llvm::Value*> multi_index = in EmitOperandArrayLoopNest() local
104 std::vector<llvm::Value*> multi_index = multidim(); in AddOffsetToDim() local
244 std::vector<llvm::Value*> multi_index(operand_shape.dimensions_size()); in SourceIndexOfBitcast() local
97 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()129 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()161 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()649 result.Populate<complex64>([&](absl::Span<const int64> multi_index) { in HandleComplex()657 result.Populate<complex128>([&](absl::Span<const int64> multi_index) { in HandleComplex()
2054 std::vector<llvm::Value*> multi_index = padded_index.multidim(); in EmitElementalPad() local
2084 auto* multi_index = absl::get_if<absl::Span<const int64>>(&index); in SetValueInLiteralHelper() local