Searched refs:BumpIndices (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | index_util_test.cc | 145 EXPECT_TRUE(IndexUtil::BumpIndices(shape, &indices)); in TEST()
147 EXPECT_TRUE(IndexUtil::BumpIndices(shape, &indices)); in TEST()
149 EXPECT_TRUE(IndexUtil::BumpIndices(shape, &indices)); in TEST()
151 EXPECT_FALSE(IndexUtil::BumpIndices(shape, &indices)); in TEST()
|
| D | index_util.h | 61 static bool BumpIndices(const Shape& shape,
|
| D | index_util.cc | 121 /* static */ bool IndexUtil::BumpIndices( in BumpIndices() function in xla::IndexUtil
|
| D | literal_util.cc | 311 } while (IndexUtil::BumpIndices(dest_shape, &index)); in CopyElementsBetween()
1284 } while (IndexUtil::BumpIndices(shape(), &indices)); in EachCellAsString()
|
| D | literal_util.h | 1159 } while (IndexUtil::BumpIndices(shape(), &indices)); in EachCell()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 194 } while (IndexUtil::BumpIndices(window_shape, &window_index)); in IterateThroughWindow()
1014 } while (IndexUtil::BumpIndices(window_shape, &rhs_spatial_index)); in HandleConvolution()
1623 } while (IndexUtil::BumpIndices(source->shape(), &source_index)); in HandleSelectAndScatter()
|