Searched refs:inverse_dimensions (Results 1 – 2 of 2) sorted by relevance
1158 auto inverse_dimensions = InversePermutation(user->dimensions()); in ChooseOutputLayoutFromOperandLayout() local1161 int64 user_dim = inverse_dimensions[operand_dim]; in ChooseOutputLayoutFromOperandLayout()
1952 std::vector<int64> inverse_dimensions(hlo->shape().rank()); in HandleTranspose() local1954 inverse_dimensions[hlo->dimensions(i)] = i; in HandleTranspose()1957 hlo_sharding_util::TransposeSharding(sharding, inverse_dimensions); in HandleTranspose()