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

Function ConvertAddNOperator

tensorflow/lite/toco/export_tensorflow.cc:687–698  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

685}
686
687void ConvertAddNOperator(const Model& model, const AddNOperator& src_op,
688 GraphDef* tensorflow_graph) {
689 tensorflow::NodeDef* add_op = tensorflow_graph->add_node();
690 add_op->set_op("AddN");
691 add_op->set_name(src_op.outputs[0]);
692 for (const auto& input : src_op.inputs) {
693 *add_op->add_input() = input;
694 }
695 (*add_op->mutable_attr())["N"].set_i(src_op.inputs.size());
696 (*add_op->mutable_attr())["T"].set_type(
697 GetTensorFlowDataType(model, src_op.outputs[0]));
698}
699
700void ConvertMulOperator(const Model& model, const MulOperator& src_op,
701 GraphDef* tensorflow_graph) {

Callers 1

ConvertOperatorFunction · 0.85

Calls 5

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

Tested by

no test coverage detected