Home
last modified time | relevance | path

Searched defs:multi_index (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_comparison.cc53 absl::Span<const int64_t> multi_index) { in CompareFloatsBitwiseEqual()
64 absl::Span<const int64_t> multi_index) { in CompareEqual()
72 absl::Span<const int64_t> multi_index) { in CompareEqual()
77 absl::Span<const int64_t> multi_index) { in CompareEqual()
83 absl::Span<const int64_t> multi_index) { in CompareEqual()
88 absl::Span<const int64_t> multi_index) { in CompareEqual()
93 absl::Span<const int64_t> multi_index) { in CompareEqual()
99 absl::Span<const int64_t> multi_index) { in CompareEqual()
106 absl::Span<const int64_t> multi_index) { in MakeBitwiseErrorStatus()
121 absl::Span<const int64_t> multi_index) { in MakeErrorStatus()
[all …]
Dliteral_util.cc243 absl::Span<const int64_t> multi_index) { in GetElementAtIndexImpl()
250 absl::Span<const int64_t> multi_index) { in GetElementAtIndexImpl()
265 absl::Span<const int64_t> multi_index, in SetScalarAtIndexImpl()
472 const LiteralBase& literal, absl::Span<const int64_t> multi_index) { in GetScalarLiteral()
478 MutableLiteralBase& literal, absl::Span<const int64_t> multi_index, in SetScalarLiteral()
583 absl::Span<const int64_t> multi_index) { in MultiIndexAsString()
Dindex_util.cc29 const Shape& shape, absl::Span<const int64_t> multi_index) { in MultidimensionalIndexToLinearIndex()
104 std::vector<int64_t> multi_index(shape.dimensions_size()); in LinearIndexToMultidimensionalIndex() local
Dliteral.h154 IsEqualAt(absl::Span<const int64_t> multi_index, T value) const { in IsEqualAt()
162 bool IsEqualAt(absl::Span<const int64_t> multi_index, in IsEqualAt()
1221 NativeT LiteralBase::Piece::Get(absl::Span<const int64_t> multi_index) const { in Get()
1228 void LiteralBase::Piece::Set(absl::Span<const int64_t> multi_index, in Set()
1247 inline NativeT LiteralBase::Get(absl::Span<const int64_t> multi_index, in Get()
1253 inline NativeT LiteralBase::Get(absl::Span<const int64_t> multi_index) const { in Get()
1258 inline void MutableLiteralBase::Set(absl::Span<const int64_t> multi_index, in Set()
1265 inline void MutableLiteralBase::Set(absl::Span<const int64_t> multi_index, in Set()
Dindex_util_test.cc40 std::vector<int64_t> multi_index = in TEST() local
133 std::vector<int64_t> multi_index = in TEST() local
Dliteral.cc381 absl::Span<const int64_t> multi_index) { in CopySliceFromInternal()
1126 std::string LiteralBase::GetAsString(absl::Span<const int64_t> multi_index, in GetAsString()
1239 absl::Span<const int64_t> multi_index, int64_t value) { in SetIntegralAsS64()
1267 Status MutableLiteralBase::SetFromDouble(absl::Span<const int64_t> multi_index, in SetFromDouble()
1834 std::vector<int64_t> multi_index; in EqualElements() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_loop.cc246 std::vector<llvm::Value*> multi_index(shape.dimensions_size()); in AddLoopsForShapeOnDimensions() local
276 std::vector<llvm::Value*> multi_index = in EmitOperandArrayLoopNest() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_simplify_padding.cc163 absl::InlinedVector<int64_t, 5> multi_index; in NumTrailingZeroOutputFeatures() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc112 result.Populate<bool>([&](absl::Span<const int64_t> multi_index) { in Compare()
144 result.Populate<bool>([&](absl::Span<const int64_t> multi_index) { in Compare()
176 result.Populate<bool>([&](absl::Span<const int64_t> multi_index) { in Compare()
1372 [&](absl::Span<const int64_t> multi_index) { in HandleComplex()
1380 [&](absl::Span<const int64_t> multi_index) { in HandleComplex()
Delemental_ir_emitter.cc2306 std::vector<llvm::Value*> multi_index = padded_index.multidim(); in EmitElementalPad() local
/external/rust/crates/syn/src/
Dexpr.rs2897 fn multi_index(e: &mut Expr, dot_token: &mut Token![.], float: LitFloat) -> Result<bool> { in multi_index() function