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

Function ConvertBiasAddOperator

tensorflow/lite/toco/import_tensorflow.cc:994–1009  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

992}
993
994tensorflow::Status ConvertBiasAddOperator(
995 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags,
996 const ModelFlags& model_flags, Model* model) {
997 CHECK_EQ(node.op(), "BiasAdd");
998 TF_QCHECK_OK(CheckInputsCount(node, tf_import_flags, 2));
999
1000 const auto& input_name = node.input(0);
1001 const auto& bias_name = node.input(1);
1002 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_FLOAT);
1003 auto* biasadd = new AddOperator;
1004 biasadd->inputs.push_back(input_name);
1005 biasadd->inputs.push_back(bias_name);
1006 biasadd->outputs.push_back(node.name());
1007 model->operators.emplace_back(biasadd);
1008 return tensorflow::Status::OK();
1009}
1010
1011tensorflow::Status ConvertRandomUniform(
1012 const NodeDef& node, const TensorFlowImportFlags& tf_import_flags,

Callers

nothing calls this directly

Calls 7

CheckInputsCountFunction · 0.85
GetDataTypeAttrFunction · 0.85
nameMethod · 0.65
opMethod · 0.45
inputMethod · 0.45
push_backMethod · 0.45
emplace_backMethod · 0.45

Tested by

no test coverage detected