Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3423 std::vector<int32_t> pad_index; in ConvertPad() local
3426 pad_index.push_back(i); in ConvertPad()
3431 if (pad_index.empty()) { in ConvertPad()
3437 if (pad_index.size() > 2) { in ConvertPad()
3443 if (pad_index[0] == 0) { in ConvertPad()
3450 if (pad_index.size() == 2 && pad_index[0] == 0 && pad_index[1] == 3) { in ConvertPad()
3460 std::vector<int32_t> permuted_pad_index(pad_index); in ConvertPad()
3461 if (pad_index[0] == 1) { in ConvertPad()
3468 for (size_t i = 0; i < pad_index.size(); i++) { in ConvertPad()
3469 int index = pad_index[i]; in ConvertPad()