Searched defs:multi_index (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_comparison.cc | 53 absl::Span<const int64> multi_index) { in CompareFloatsBitwiseEqual()
64 absl::Span<const int64> multi_index) { in CompareEqual()
72 absl::Span<const int64> multi_index) { in CompareEqual()
77 absl::Span<const int64> multi_index) { in CompareEqual()
82 absl::Span<const int64> multi_index) { in CompareEqual()
87 absl::Span<const int64> multi_index) { in CompareEqual()
92 absl::Span<const int64> multi_index) { in CompareEqual()
98 absl::Span<const int64> multi_index) { in CompareEqual()
105 absl::Span<const int64> multi_index) { in MakeBitwiseErrorStatus()
120 absl::Span<const int64> multi_index) { in MakeErrorStatus()
[all …]
|
| D | literal.h | 153 IsEqualAt(absl::Span<const int64> multi_index, T value) const { in IsEqualAt()
161 bool IsEqualAt(absl::Span<const int64> multi_index, complex128 value) const { in IsEqualAt()
936 NativeT LiteralBase::Piece::Get(absl::Span<const int64> multi_index) const { in Get()
943 void LiteralBase::Piece::Set(absl::Span<const int64> multi_index, in Set()
962 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index, in Get()
968 inline NativeT LiteralBase::Get(absl::Span<const int64> multi_index) const { in Get()
973 inline void MutableLiteralBase::Set(absl::Span<const int64> multi_index, in Set()
980 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 | 279 absl::Span<const int64> multi_index) { in CopySliceFromInternal()
986 string LiteralBase::GetAsString(absl::Span<const int64> multi_index, in GetAsString()
1119 Status MutableLiteralBase::SetIntegralAsS64(absl::Span<const int64> multi_index, in SetIntegralAsS64()
1148 Status MutableLiteralBase::SetFromDouble(absl::Span<const int64> multi_index, in SetFromDouble()
1690 std::vector<int64> multi_index; in EqualElements() local
|
| D | literal_util.cc | 576 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 | 100 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()
132 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()
164 result.Populate<bool>([&](absl::Span<const int64> multi_index) { in Compare()
721 result.Populate<complex64>([&](absl::Span<const int64> multi_index) { in HandleComplex()
729 result.Populate<complex128>([&](absl::Span<const int64> multi_index) { in HandleComplex()
|
| D | elemental_ir_emitter.cc | 2299 std::vector<llvm::Value*> multi_index = padded_index.multidim(); in EmitElementalPad() local
|
| /external/rust/crates/syn/src/ |
| D | expr.rs | 2864 fn multi_index(e: &mut Expr, dot_token: &mut Token![.], float: LitFloat) -> Result<bool> { in multi_index() function
|