Searched defs:multi_index (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_comparison.cc | 51 absl::Span<const int64> multi_index) { in CompareFloatsBitwiseEqual()
62 absl::Span<const int64> multi_index) { in CompareEqual()
70 absl::Span<const int64> multi_index) { in CompareEqual()
75 absl::Span<const int64> multi_index) { in CompareEqual()
80 absl::Span<const int64> multi_index) { in CompareEqual()
85 absl::Span<const int64> multi_index) { in CompareEqual()
90 absl::Span<const int64> multi_index) { in CompareEqual()
96 absl::Span<const int64> multi_index) { in CompareEqual()
103 absl::Span<const int64> multi_index) { in MakeBitwiseErrorStatus()
118 absl::Span<const int64> multi_index) { in MakeErrorStatus()
[all …]
|
| D | literal.h | 133 IsEqualAt(absl::Span<const int64> multi_index, T value) const { in IsEqualAt()
141 bool IsEqualAt(absl::Span<const int64> multi_index, complex128 value) const { in IsEqualAt()
857 NativeT LiteralBase::Piece::Get(absl::Span<const int64> multi_index) const { in Get()
864 void LiteralBase::Piece::Set(absl::Span<const int64> multi_index, in Set()
883 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index, in Get()
889 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index) const { in Get()
894 inline void MutableLiteralBase::Set(absl::Span<const int64> multi_index, in Set()
901 inline void MutableLiteralBase::Set(absl::Span<const int64> multi_index, in Set()
|
| D | index_util.cc | 29 const Shape& shape, absl::Span<const int64> multi_index) { in MultidimensionalIndexToLinearIndex()
104 std::vector<int64> multi_index(shape.dimensions_size()); in LinearIndexToMultidimensionalIndex() local
|
| D | index_util_test.cc | 40 std::vector<int64> multi_index = in TEST() local
133 std::vector<int64> multi_index = in TEST() local
|
| D | literal.cc | 208 absl::Span<const int64> multi_index) { in CopySliceFromInternal()
782 string LiteralBase::GetAsString(absl::Span<const int64> multi_index, in GetAsString()
909 Status MutableLiteralBase::SetIntegralAsS64(absl::Span<const int64> multi_index, in SetIntegralAsS64()
938 Status MutableLiteralBase::SetFromDouble(absl::Span<const int64> multi_index, in SetFromDouble()
1372 std::vector<int64> multi_index; in EqualElements() local
|
| D | literal_util.cc | 510 absl::Span<const int64> multi_index) { in MultiIndexAsString()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_loop.cc | 245 std::vector<llvm::Value*> multi_index(shape.dimensions_size()); in AddLoopsForShapeOnDimensions() local
275 std::vector<llvm::Value*> multi_index = in EmitOperandArrayLoopNest() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 99 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()
131 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()
163 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()
674 result.Populate<complex64>([&](absl::Span<const int64> multi_index) { in HandleComplex()
682 result.Populate<complex128>([&](absl::Span<const int64> multi_index) { in HandleComplex()
|
| D | elemental_ir_emitter.cc | 1998 std::vector<llvm::Value*> multi_index = padded_index.multidim(); in EmitElementalPad() local
|