Home
last modified time | relevance | path

Searched refs:LinearIndexToMultidimensionalIndex (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dindex_util_test.cc41 IndexUtil::LinearIndexToMultidimensionalIndex(vector_shape, 42); in TEST()
59 IndexUtil::LinearIndexToMultidimensionalIndex(matrix_shape_01, 53)); in TEST()
75 IndexUtil::LinearIndexToMultidimensionalIndex(matrix_shape_10, 65)); in TEST()
134 IndexUtil::LinearIndexToMultidimensionalIndex(shape, linear_index); in TEST()
H A Dindex_util.h43 static std::vector<int64_t> LinearIndexToMultidimensionalIndex(
H A Dindex_util.cc92 /* static */ std::vector<int64_t> IndexUtil::LinearIndexToMultidimensionalIndex( in LinearIndexToMultidimensionalIndex() function in xla::IndexUtil
H A Dliteral_util.cc410 IndexUtil::LinearIndexToMultidimensionalIndex(literal.shape(), i); in ReshapeSlice()
412 IndexUtil::LinearIndexToMultidimensionalIndex(shape_with_layout, i); in ReshapeSlice()
H A Dliteral_comparison.cc315 IndexUtil::LinearIndexToMultidimensionalIndex(shape, in ToString()
H A Dshape_util.cc1483 IndexUtil::LinearIndexToMultidimensionalIndex(output_shape_dim0_major, in ReshapeIsBitcast()
H A Dliteral.cc1482 std::vector<int64_t> indices = IndexUtil::LinearIndexToMultidimensionalIndex( in EachCellAsString()