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.cc51 const auto* max_op = op_0->type == OperatorType::kMaximum ? op_0 : op_1; in Run() local
53 if (min_op->inputs.size() != 2 || max_op->inputs.size() != 2) { in Run()
56 if (min_op->outputs.size() != 1 || max_op->outputs.size() != 1) { in Run()
64 GetSingleScalarInputIndexOfBinaryOp(model, max_op, -1.0f); in Run()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc763 tensorflow::NodeDef* max_op = tensorflow_graph->add_node(); in ConvertRelu1Operator() local
764 max_op->set_op("Maximum"); in ConvertRelu1Operator()
765 max_op->set_name(max_output); in ConvertRelu1Operator()
766 *max_op->add_input() = src_op.inputs[0]; in ConvertRelu1Operator()
767 *max_op->add_input() = max_bounds; in ConvertRelu1Operator()
768 (*max_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertRelu1Operator()
1898 tensorflow::NodeDef* max_op = tensorflow_graph->add_node(); in ConvertTensorFlowMaximumOperator() local
1899 max_op->set_op("Maximum"); in ConvertTensorFlowMaximumOperator()
1900 max_op->set_name(src_op.outputs[0]); in ConvertTensorFlowMaximumOperator()
1902 *max_op->add_input() = src_op.inputs[0]; in ConvertTensorFlowMaximumOperator()
[all …]