Searched refs:permutation_size (Results 1 – 4 of 4) sorted by relevance
568 int permutation_size = permutation.size(); in PermuteSingle() local569 if (values->size() != permutation_size) { in PermuteSingle()573 permutation_size, " @ ", location)); in PermuteSingle()590 int permutation_size = permutation.size(); in PermuteDouble() local591 if (values->size() != permutation_size * 2) { in PermuteDouble()595 permutation_size, " @ ", location)); in PermuteDouble()
1055 const int permutation_size = permutation.size(); in IsValidConstPermTransposeNode() local1056 if (tensor.NumElements() != permutation_size) { in IsValidConstPermTransposeNode()1061 for (int i = 0; i < permutation_size; i++) { in IsValidConstPermTransposeNode()
2150 int permutation_size = permutation.size(); in RemoveShuffleOrTranspose() local2151 if (permutation_size != shape.dim_size()) { in RemoveShuffleOrTranspose()
1153 const int permutation_size = permutation.size(); in RenumberArguments() local1154 if (index < 0 || index >= permutation_size) { in RenumberArguments()