Searched refs:perm_array_val (Results 1 – 1 of 1) sorted by relevance
166 std::vector<int32> perm_array_val(dims); in GetTransposePerm() local168 perm_array_val[i] = i; in GetTransposePerm()170 perm_array_val[dims - 2] = dims - 1; in GetTransposePerm()171 perm_array_val[dims - 1] = dims - 2; in GetTransposePerm()172 return perm_array_val; in GetTransposePerm()176 const std::vector<int32>& perm_array_val) { in GetTransposeShape() argument180 output_shape[i] = input_shape.dims(perm_array_val[i]); in GetTransposeShape()