Home
last modified time | relevance | path

Searched refs:MultidimensionalIndexToLinearIndex (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dindex_util_test.cc39 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 Dliteral.cc382 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 Dindex_util.h37 static int64_t MultidimensionalIndexToLinearIndex(
H A Dindex_util.cc28 /* static */ int64_t IndexUtil::MultidimensionalIndexToLinearIndex( in MultidimensionalIndexToLinearIndex() function in xla::IndexUtil
H A Dliteral.h1223 return data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex( in Get()
1231 data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex( in Set()
1397 IndexUtil::MultidimensionalIndexToLinearIndex(shape(), indexes); in PopulateInternal()
H A Dshape_util.cc1479 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 Dliteral_comparison.cc536 IndexUtil::MultidimensionalIndexToLinearIndex( in CompareLiteralsSlow()
H A Dliteral_test.cc1259 return IndexUtil::MultidimensionalIndexToLinearIndex(literal.shape(), in TEST_F()
1302 return IndexUtil::MultidimensionalIndexToLinearIndex(literal.shape(), in TEST_F()