Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1428 const std::vector<std::pair<string, bool>>& inputs_is_weight) { in CheckInputsWeights() argument
1431 if (inputs.size() != inputs_is_weight.size()) { in CheckInputsWeights()
1434 inputs_is_weight.size(), ", at ", node_def.name()); in CheckInputsWeights()
1437 if (inputs_is_weight[i].second && inputs.at(i).is_tensor()) { in CheckInputsWeights()
1438 return errors::Unimplemented("The input \"", inputs_is_weight[i].first, in CheckInputsWeights()
1446 if (!inputs_is_weight[i].second && inputs.at(i).is_weights()) { in CheckInputsWeights()
1447 return errors::Unimplemented("The input \"", inputs_is_weight[i].first, in CheckInputsWeights()