Searched refs:nb_dims (Results 1 – 1 of 1) sorted by relevance
4832 const int nb_dims = in ConvertPad() local4837 if (nb_dims < 4) { in ConvertPad()4849 if (pads.shape_.d[0] != nb_dims || pads.shape_.d[1] != 2) { in ConvertPad()4862 for (int i = 0; i < nb_dims; i++) { in ConvertPad()4893 std::vector<int> transpose_idx(nb_dims); in ConvertPad()4897 std::vector<int> trt_pad_index{nb_dims - 2, nb_dims - 1}; in ConvertPad()4907 if (tf_pad_index.size() == 1 && tf_pad_index[0] == nb_dims - 1) { in ConvertPad()4910 trt_pad_index[0] = nb_dims - 1; in ConvertPad()4913 if (tf_pad_index.size() == 2 && tf_pad_index[1] == nb_dims - 2) { in ConvertPad()