Home
last modified time | relevance | path

Searched refs:max_op (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_relu1.cc80 const auto* max_op = op_0->type == OperatorType::kMaximum ? op_0 : op_1; in Run() local
82 if (min_op->inputs.size() != 2 || max_op->inputs.size() != 2) { in Run()
85 if (min_op->outputs.size() != 1 || max_op->outputs.size() != 1) { in Run()
93 GetSingleScalarInputIndexOfBinaryOp(model, max_op, -1.0f); in Run()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc758 tensorflow::NodeDef* max_op = tensorflow_graph->add_node(); in ConvertRelu1Operator() local
759 max_op->set_op("Maximum"); in ConvertRelu1Operator()
760 max_op->set_name(max_output); in ConvertRelu1Operator()
761 *max_op->add_input() = src_op.inputs[0]; in ConvertRelu1Operator()
762 *max_op->add_input() = max_bounds; in ConvertRelu1Operator()
763 (*max_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertRelu1Operator()
1864 tensorflow::NodeDef* max_op = tensorflow_graph->add_node(); in ConvertTensorFlowMaximumOperator() local
1865 max_op->set_op("Maximum"); in ConvertTensorFlowMaximumOperator()
1866 max_op->set_name(src_op.outputs[0]); in ConvertTensorFlowMaximumOperator()
1868 *max_op->add_input() = src_op.inputs[0]; in ConvertTensorFlowMaximumOperator()
[all …]