Searched refs:BumpIndices (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | index_util_test.cc | 144 EXPECT_TRUE(IndexUtil::BumpIndices(shape, absl::MakeSpan(indices))); in TEST()
146 EXPECT_TRUE(IndexUtil::BumpIndices(shape, absl::MakeSpan(indices))); in TEST()
148 EXPECT_TRUE(IndexUtil::BumpIndices(shape, absl::MakeSpan(indices))); in TEST()
150 EXPECT_FALSE(IndexUtil::BumpIndices(shape, absl::MakeSpan(indices))); in TEST()
|
| D | index_util.h | 62 static bool BumpIndices(const Shape& shape, absl::Span<int64> indices);
|
| D | index_util.cc | 116 /* static */ bool IndexUtil::BumpIndices(const Shape& shape, in BumpIndices() function in xla::IndexUtil
|
| D | literal.h | 987 } while (IndexUtil::BumpIndices(shape(), absl::MakeSpan(indices))); in EachCell()
|
| D | literal.cc | 382 } while (IndexUtil::BumpIndices(dest_shape, absl::MakeSpan(index))); in CopyElementsBetween()
1230 } while (IndexUtil::BumpIndices(shape(), absl::MakeSpan(indices))); in EachCellAsString()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1191 } while (IndexUtil::BumpIndices(window_shape, in HandleConvolution()
1955 IndexUtil::BumpIndices(source->shape(), absl::MakeSpan(source_index))); in HandleSelectAndScatter()
2854 IndexUtil::BumpIndices(window_shape, absl::MakeSpan(window_index))); in IterateThroughWindow()
|