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

Function ConvertMulOperator

tensorflow/lite/toco/export_tensorflow.cc:700–710  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

698}
699
700void ConvertMulOperator(const Model& model, const MulOperator& src_op,
701 GraphDef* tensorflow_graph) {
702 tensorflow::NodeDef* mul_op = tensorflow_graph->add_node();
703 mul_op->set_op("Mul");
704 mul_op->set_name(src_op.outputs[0]);
705 CHECK_EQ(src_op.inputs.size(), 2);
706 *mul_op->add_input() = src_op.inputs[0];
707 *mul_op->add_input() = src_op.inputs[1];
708 (*mul_op->mutable_attr())["T"].set_type(
709 GetTensorFlowDataType(model, src_op.outputs[0]));
710}
711
712void ConvertDivOperator(const Model& model, const DivOperator& src_op,
713 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