Searched refs:inputs_is_weight (Results 1 – 1 of 1) sorted by relevance
2086 const std::vector<std::pair<string, bool>>& inputs_is_weight) { in CheckInputsWeights() argument2088 expected_inputs.reserve(inputs_is_weight.size()); in CheckInputsWeights()2090 inputs_is_weight.begin(), inputs_is_weight.end(), in CheckInputsWeights()4736 std::vector<std::pair<string, TrtInputArg>> inputs_is_weight; in ConvertPack() local4737 inputs_is_weight.reserve(num_inputs); in ConvertPack()4739 inputs_is_weight.push_back({StrCat("values_", i), expected_arg}); in ConvertPack()4741 TF_RETURN_IF_ERROR(CheckInputsWeights(*params, inputs_is_weight)); in ConvertPack()