Searched refs:TrtInputArg (Results 1 – 1 of 1) sorted by relevance
2046 enum class TrtInputArg { kTensor = 1, kWeight = 2, kBoth = 3 }; enum2052 const std::vector<std::pair<string, TrtInputArg>>& expected_inputs) { in CheckInputsWeights()2061 if (expected_inputs[i].second == TrtInputArg::kWeight && in CheckInputsWeights()2071 if (expected_inputs[i].second == TrtInputArg::kTensor && in CheckInputsWeights()2087 std::vector<std::pair<string, TrtInputArg>> expected_inputs; in CheckInputsWeights()2093 x.first, x.second ? TrtInputArg::kWeight : TrtInputArg::kTensor); in CheckInputsWeights()2535 CheckInputsWeights(*params, {{"input", TrtInputArg::kBoth}})); in ConvertShape()4733 TrtInputArg expected_arg = in ConvertPack()4734 params->use_implicit_batch ? TrtInputArg::kTensor : TrtInputArg::kBoth; in ConvertPack()4736 std::vector<std::pair<string, TrtInputArg>> inputs_is_weight; in ConvertPack()[all …]