Searched refs:input_tensor_rank (Results 1 – 2 of 2) sorted by relevance
99 int input_tensor_rank = input0_tensor_shape.size(); in convertPackOp() local110 if (next_tensor_shape.size() != input_tensor_rank) { in convertPackOp()124 if (input_tensor_rank == 0) { in convertPackOp()145 if (axis > (input_tensor_rank + 1)) { in convertPackOp()155 if (output_shape_vals.size() != (input_tensor_rank + 1)) { in convertPackOp()176 if (axis == 0 && input_tensor_rank == 0) { in convertPackOp()180 } else if (axis == input_tensor_rank) { in convertPackOp()188 for (int d = 0; d < input_tensor_rank; d++) { in convertPackOp()207 if (input_tensor_rank == 0) { in convertPackOp()211 for (int i = 0; i < input_tensor_rank; i++) { in convertPackOp()[all …]
527 size_t input_tensor_rank = %input[0].rank530 if (input_tensor_rank == 0) {537 for (int i = 0; i < input_tensor_rank; i++) {551 if (input_tensor_rank == 0) {558 if (concat_axis == input_tensor_rank) {