Searched refs:LinearIndexToMultidimensionalIndex (Results 1 – 7 of 7) sorted by relevance
41 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()
44 static std::vector<int64> LinearIndexToMultidimensionalIndex(
92 /* static */ std::vector<int64> IndexUtil::LinearIndexToMultidimensionalIndex( in LinearIndexToMultidimensionalIndex() function in xla::IndexUtil
314 IndexUtil::LinearIndexToMultidimensionalIndex(literal.shape(), i); in ReshapeSlice()316 IndexUtil::LinearIndexToMultidimensionalIndex(shape_with_layout, i); in ReshapeSlice()
297 IndexUtil::LinearIndexToMultidimensionalIndex(shape, in ToString()
1243 IndexUtil::LinearIndexToMultidimensionalIndex(output_shape_dim0_major, in ReshapeIsBitcast()
1226 std::vector<int64> indices = IndexUtil::LinearIndexToMultidimensionalIndex( in EachCellAsString()