Searched refs:CheckSimpleOperator (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/ |
| H A D | operator_test.cc | 91 void CheckSimpleOperator(const std::string& name, OperatorType type) { in CheckSimpleOperator() function in toco::tflite::__anon5be81a820111::OperatorTest
115 CheckSimpleOperator<FloorOperator>("FLOOR", OperatorType::kFloor); in TEST_F()
116 CheckSimpleOperator<CeilOperator>("CEIL", OperatorType::kCeil); in TEST_F()
117 CheckSimpleOperator<EluOperator>("ELU", OperatorType::kElu); in TEST_F()
118 CheckSimpleOperator<RoundOperator>("ROUND", OperatorType::kRound); in TEST_F()
119 CheckSimpleOperator<ReluOperator>("RELU", OperatorType::kRelu); in TEST_F()
120 CheckSimpleOperator<Relu1Operator>("RELU_N1_TO_1", OperatorType::kRelu1); in TEST_F()
121 CheckSimpleOperator<Relu6Operator>("RELU6", OperatorType::kRelu6); in TEST_F()
122 CheckSimpleOperator<LogisticOperator>("LOGISTIC", OperatorType::kLogistic); in TEST_F()
123 CheckSimpleOperator<TanhOperator>("TANH", OperatorType::kTanh); in TEST_F()
[all …]
|