Searched refs:MultidimensionalIndexToLinearIndex (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | index_util_test.cc | 40 IndexUtil::MultidimensionalIndexToLinearIndex(vector_shape, {42})); in TEST()
53 EXPECT_EQ(0, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01, in TEST()
55 EXPECT_EQ(199, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01, in TEST()
57 EXPECT_EQ(53, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_01, in TEST()
69 EXPECT_EQ(0, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_10, in TEST()
71 EXPECT_EQ(199, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_10, in TEST()
73 EXPECT_EQ(65, IndexUtil::MultidimensionalIndexToLinearIndex(matrix_shape_10, in TEST()
86 EXPECT_EQ(1957, IndexUtil::MultidimensionalIndexToLinearIndex(shape_210, in TEST()
88 EXPECT_EQ(5277, IndexUtil::MultidimensionalIndexToLinearIndex(shape_210, in TEST()
99 EXPECT_EQ(1945, IndexUtil::MultidimensionalIndexToLinearIndex(shape_120, in TEST()
[all …]
|
| D | index_util.h | 37 static int64 MultidimensionalIndexToLinearIndex(
|
| D | index_util.cc | 29 /* static */ int64 IndexUtil::MultidimensionalIndexToLinearIndex( in MultidimensionalIndexToLinearIndex() function in xla::IndexUtil
|
| D | literal_util.h | 832 return data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex( in Get()
840 data<NativeT>()[IndexUtil::MultidimensionalIndexToLinearIndex( in Set()
1268 IndexUtil::MultidimensionalIndexToLinearIndex(shape(), indexes); in Populate()
|
| D | shape_util.cc | 1248 IndexUtil::MultidimensionalIndexToLinearIndex(input_shape_dim0_major, in ReshapeIsBitcast()
1256 if (IndexUtil::MultidimensionalIndexToLinearIndex(input_shape, in ReshapeIsBitcast()
1258 IndexUtil::MultidimensionalIndexToLinearIndex(output_shape, in ReshapeIsBitcast()
|
| D | literal_util.cc | 199 return IndexUtil::MultidimensionalIndexToLinearIndex(shape, multi_index); in CopySliceFromInternal()
309 dest[IndexUtil::MultidimensionalIndexToLinearIndex(dest_shape, index)] = in CopyElementsBetween()
310 src[IndexUtil::MultidimensionalIndexToLinearIndex(src_shape, index)]; in CopyElementsBetween()
|
| D | literal_util_test.cc | 1055 return IndexUtil::MultidimensionalIndexToLinearIndex(literal->shape(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | literal_test_util.cc | 562 IndexUtil::MultidimensionalIndexToLinearIndex(actual.shape(), in ExpectLiteralsNear()
|