Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc4861 std::vector<int32_t> tf_pad_index; in ConvertPad() local
4864 tf_pad_index.push_back(i); in ConvertPad()
4869 if (tf_pad_index.empty()) { in ConvertPad()
4877 if (tf_pad_index.size() > 2) { in ConvertPad()
4883 if (params->use_implicit_batch && tf_pad_index[0] == 0) { in ConvertPad()
4907 if (tf_pad_index.size() == 1 && tf_pad_index[0] == nb_dims - 1) { in ConvertPad()
4913 if (tf_pad_index.size() == 2 && tf_pad_index[1] == nb_dims - 2) { in ConvertPad()
4928 for (int i = 0; i < tf_pad_index.size(); i++) { in ConvertPad()
4929 const int tf_index = tf_pad_index[i]; in ConvertPad()