Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc1671 const std::vector<std::vector<int>>& exp_out_dims) { in RunValidationAndConversion() argument
1681 for (int i = 0; i < exp_out_dims.size(); i++) { in RunValidationAndConversion()
1686 if (!exp_out_dims[i].empty()) { in RunValidationAndConversion()
1688 auto out_dims = std::vector<int>(exp_out_dims[i].begin() + 1, in RunValidationAndConversion()
1689 exp_out_dims[i].end()); in RunValidationAndConversion()