MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / CheckInputsCount

Function CheckInputsCount

tensorflow/lite/toco/import_tensorflow.cc:518–527  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

516}
517
518tensorflow::Status CheckInputsCount(
519 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags,
520 int expected_input_count) {
521 if (GetInputsCount(node, tf_import_flags) != expected_input_count) {
522 return tensorflow::errors::FailedPrecondition(
523 node.op(), " node expects ", expected_input_count,
524 " input(s) other than control dependencies: ", node.DebugString());
525 }
526 return tensorflow::Status::OK();
527}
528
529template <ArrayDataType T>
530string CreateConstArray(Model* model, string const& name,

Callers 15

ConvertConvOperatorFunction · 0.85
ConvertBiasAddOperatorFunction · 0.85
ConvertRandomUniformFunction · 0.85
ConvertSqueezeOperatorFunction · 0.85
ConvertSplitOperatorFunction · 0.85
ConvertSplitVOperatorFunction · 0.85
ConvertSwitchOperatorFunction · 0.85
ConvertSoftmaxOperatorFunction · 0.85

Calls 4

GetInputsCountFunction · 0.85
FailedPreconditionFunction · 0.85
opMethod · 0.45
DebugStringMethod · 0.45

Tested by

no test coverage detected