Searched refs:inputs_kinds (Results 1 – 1 of 1) sorted by relevance
5113 std::vector<std::pair<string, TrtInputArg>> inputs_kinds; in ConvertConcat() local5115 inputs_kinds.reserve(num_inputs); in ConvertConcat()5117 inputs_kinds.push_back({StrCat("values_", i), expected_input}); in ConvertConcat()5119 inputs_kinds.push_back({"axis", TrtInputArg::kWeight}); in ConvertConcat()5120 TF_RETURN_IF_ERROR(CheckInputsWeights(*params, inputs_kinds)); in ConvertConcat()