Searched refs:transpose_idx (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 4893 std::vector<int> transpose_idx(nb_dims); in ConvertPad() local
4894 std::iota(transpose_idx.begin(), transpose_idx.end(), 0); in ConvertPad()
4936 std::swap(transpose_idx[tf_index], transpose_idx[trt_index]); in ConvertPad()
4942 tensor, transpose_idx, &tensor, node_def, "to_pad")); in ConvertPad()
4954 output_tensor, transpose_idx, &output_tensor, node_def, "from_pad")); in ConvertPad()
|