Searched defs:multi_index (Results 1 – 10 of 10) 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 | 149 IsEqualAt(absl::Span<const int64> multi_index, T value) const { in IsEqualAt()
157 bool IsEqualAt(absl::Span<const int64> multi_index, complex128 value) const { in IsEqualAt()
920 NativeT LiteralBase::Piece::Get(absl::Span<const int64> multi_index) const { in Get()
927 void LiteralBase::Piece::Set(absl::Span<const int64> multi_index, in Set()
946 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index, in Get()
952 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index) const { in Get()
957 inline void MutableLiteralBase::Set(absl::Span<const int64> multi_index, in Set()
964 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 | 276 absl::Span<const int64> multi_index) { in CopySliceFromInternal()
972 string LiteralBase::GetAsString(absl::Span<const int64> multi_index, in GetAsString()
1105 Status MutableLiteralBase::SetIntegralAsS64(absl::Span<const int64> multi_index, in SetIntegralAsS64()
1134 Status MutableLiteralBase::SetFromDouble(absl::Span<const int64> multi_index, in SetFromDouble()
1635 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()
686 result.Populate<complex64>([&](absl::Span<const int64> multi_index) { in HandleComplex()
694 result.Populate<complex128>([&](absl::Span<const int64> multi_index) { in HandleComplex()
|
| D | elemental_ir_emitter.cc | 2113 std::vector<llvm::Value*> multi_index = padded_index.multidim(); in EmitElementalPad() local
|
| /external/rust/crates/syn/src/ |
| D | expr.rs | 2836 fn multi_index(e: &mut Expr, dot_token: &mut Token![.], float: LitFloat) -> Result<bool> { in multi_index() function
|