Searched refs:MultidimensionalIndexToLinearIndex (Results 1 – 8 of 8) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
H A D | index_util_test.cc | 39 IndexUtil::MultidimensionalIndexToLinearIndex(vector_shape, {42})); in TEST() 52 EXPECT_EQ(0, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01, in TEST() 54 EXPECT_EQ(199, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01, in TEST() 56 EXPECT_EQ(53, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01, in TEST() 68 EXPECT_EQ(0, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_10, in TEST() 70 EXPECT_EQ(199, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_10, in TEST() 72 EXPECT_EQ(65, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_10, in TEST() 85 EXPECT_EQ(1957, IndexUtil::MultidimensionalIndexToLinearIndex(shape_210, in TEST() 87 EXPECT_EQ(5277, IndexUtil::MultidimensionalIndexToLinearIndex(shape_210, in TEST() 98 EXPECT_EQ(1945, IndexUtil::MultidimensionalIndexToLinearIndex(shape_120, in TEST() [all …]
|
H A D | literal.cc | 382 return IndexUtil::MultidimensionalIndexToLinearIndex(shape, multi_index); in CopySliceFromInternal() 438 IndexUtil::MultidimensionalIndexToLinearIndex(src_literal.shape(), in CopyElementFrom() 441 IndexUtil::MultidimensionalIndexToLinearIndex(shape(), dest_index); in CopyElementFrom() 564 dest[IndexUtil::MultidimensionalIndexToLinearIndex(dest_shape, index)] = in CopyElementsBetween() 565 src[IndexUtil::MultidimensionalIndexToLinearIndex(src_shape, index)]; in CopyElementsBetween() 946 int64_t dest_index = IndexUtil::MultidimensionalIndexToLinearIndex( in Broadcast() 948 int64_t source_index = IndexUtil::MultidimensionalIndexToLinearIndex( in Broadcast() 1788 data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex(dest_shape, in CopyElementsWithDynamicBound() 1790 src.data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex( in CopyElementsWithDynamicBound()
|
H A D | index_util.h | 37 static int64_t MultidimensionalIndexToLinearIndex(
|
H A D | index_util.cc | 28 /* static */ int64_t IndexUtil::MultidimensionalIndexToLinearIndex( in MultidimensionalIndexToLinearIndex() function in xla::IndexUtil
|
H A D | literal.h | 1223 return data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex( in Get() 1231 data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex( in Set() 1397 IndexUtil::MultidimensionalIndexToLinearIndex(shape(), indexes); in PopulateInternal()
|
H A D | shape_util.cc | 1479 IndexUtil::MultidimensionalIndexToLinearIndex(input_shape_dim0_major, in ReshapeIsBitcast() 1487 if (IndexUtil::MultidimensionalIndexToLinearIndex(input_shape, in ReshapeIsBitcast() 1489 IndexUtil::MultidimensionalIndexToLinearIndex(output_shape, in ReshapeIsBitcast()
|
H A D | literal_comparison.cc | 536 IndexUtil::MultidimensionalIndexToLinearIndex( in CompareLiteralsSlow()
|
H A D | literal_test.cc | 1259 return IndexUtil::MultidimensionalIndexToLinearIndex(literal.shape(), in TEST_F() 1302 return IndexUtil::MultidimensionalIndexToLinearIndex(literal.shape(), in TEST_F()
|