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

Function ConvertReluOperator

tensorflow/lite/toco/export_tensorflow.cc:724–732  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

722}
723
724void ConvertReluOperator(const Model& model, const ReluOperator& src_op,
725 GraphDef* tensorflow_graph) {
726 tensorflow::NodeDef* relu_op = tensorflow_graph->add_node();
727 relu_op->set_op("Relu");
728 relu_op->set_name(src_op.outputs[0]);
729 *relu_op->add_input() = src_op.inputs[0];
730 (*relu_op->mutable_attr())["T"].set_type(
731 GetTensorFlowDataType(model, src_op.outputs[0]));
732}
733
734void ConvertRelu1Operator(const Relu1Operator& src_op,
735 GraphDef* tensorflow_graph) {

Callers 1

ConvertOperatorFunction · 0.85

Calls 4

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

Tested by

no test coverage detected