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

Function ConvertAddOperator

tensorflow/lite/toco/export_tensorflow.cc:675–685  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

673}
674
675void ConvertAddOperator(const Model& model, const AddOperator& src_op,
676 GraphDef* tensorflow_graph) {
677 tensorflow::NodeDef* add_op = tensorflow_graph->add_node();
678 add_op->set_op("Add");
679 add_op->set_name(src_op.outputs[0]);
680 CHECK_EQ(src_op.inputs.size(), 2);
681 *add_op->add_input() = src_op.inputs[0];
682 *add_op->add_input() = src_op.inputs[1];
683 (*add_op->mutable_attr())["T"].set_type(
684 GetTensorFlowDataType(model, src_op.outputs[0]));
685}
686
687void ConvertAddNOperator(const Model& model, const AddNOperator& src_op,
688 GraphDef* tensorflow_graph) {

Callers 1

ConvertOperatorFunction · 0.85

Calls 5

set_opMethod · 0.80
GetTensorFlowDataTypeFunction · 0.70
set_nameMethod · 0.45
sizeMethod · 0.45
add_inputMethod · 0.45

Tested by

no test coverage detected