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()
43 static std::vector<int64_t> LinearIndexToMultidimensionalIndex(
92 /* static */ std::vector<int64_t> IndexUtil::LinearIndexToMultidimensionalIndex( in LinearIndexToMultidimensionalIndex() function in xla::IndexUtil
410 IndexUtil::LinearIndexToMultidimensionalIndex(literal.shape(), i); in ReshapeSlice()412 IndexUtil::LinearIndexToMultidimensionalIndex(shape_with_layout, i); in ReshapeSlice()
315 IndexUtil::LinearIndexToMultidimensionalIndex(shape, in ToString()
1483 IndexUtil::LinearIndexToMultidimensionalIndex(output_shape_dim0_major, in ReshapeIsBitcast()
1482 std::vector<int64_t> indices = IndexUtil::LinearIndexToMultidimensionalIndex( in EachCellAsString()