Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc1038 std::vector<int64> new_minor_to_major(rank); in ChooseOperandLayoutFromOutputLayout() local
1042 new_minor_to_major[i] = operand_dim; in ChooseOperandLayoutFromOutputLayout()
1044 Layout operand_layout = LayoutUtil::MakeLayout(new_minor_to_major); in ChooseOperandLayoutFromOutputLayout()
1099 std::vector<int64> new_minor_to_major(rank); in ChooseOutputLayoutFromOperandLayout() local
1104 new_minor_to_major[i] = user_dim; in ChooseOutputLayoutFromOperandLayout()
1106 Layout user_layout = LayoutUtil::MakeLayout(new_minor_to_major); in ChooseOutputLayoutFromOperandLayout()