Home
last modified time | relevance | path

Searched refs:min_op (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_relu1.cc50 const auto* min_op = op_0->type == OperatorType::kMinimum ? 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()
62 GetSingleScalarInputIndexOfBinaryOp(model, min_op, 1.0f); in Run()
69 op_0 == min_op ? min_scalar_input_index : max_scalar_input_index; in Run()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo-legalize-to-affine.mlir4 // CHECK-LABEL: func @min_op
5 func @min_op(%lhs: memref<4x3x2x1xf32>, %rhs: memref<4x3x2x1xf32>,
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc770 tensorflow::NodeDef* min_op = tensorflow_graph->add_node(); in ConvertRelu1Operator() local
771 min_op->set_op("Minimum"); in ConvertRelu1Operator()
772 min_op->set_name(src_op.outputs[0]); in ConvertRelu1Operator()
773 *min_op->add_input() = max_output; in ConvertRelu1Operator()
774 *min_op->add_input() = min_bounds; in ConvertRelu1Operator()
775 (*min_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertRelu1Operator()
1884 tensorflow::NodeDef* min_op = tensorflow_graph->add_node(); in ConvertTensorFlowMinimumOperator() local
1885 min_op->set_op("Minimum"); in ConvertTensorFlowMinimumOperator()
1886 min_op->set_name(src_op.outputs[0]); in ConvertTensorFlowMinimumOperator()
1888 *min_op->add_input() = src_op.inputs[0]; in ConvertTensorFlowMinimumOperator()
[all …]