| 110 | }; |
| 111 | |
| 112 | TEST_F(OperatorTest, SimpleOperators) { |
| 113 | CheckSimpleOperator<FloorOperator>("FLOOR", OperatorType::kFloor); |
| 114 | CheckSimpleOperator<CeilOperator>("CEIL", OperatorType::kCeil); |
| 115 | CheckSimpleOperator<EluOperator>("ELU", OperatorType::kElu); |
| 116 | CheckSimpleOperator<RoundOperator>("ROUND", OperatorType::kRound); |
| 117 | CheckSimpleOperator<ReluOperator>("RELU", OperatorType::kRelu); |
| 118 | CheckSimpleOperator<Relu1Operator>("RELU_N1_TO_1", OperatorType::kRelu1); |
| 119 | CheckSimpleOperator<Relu6Operator>("RELU6", OperatorType::kRelu6); |
| 120 | CheckSimpleOperator<LogisticOperator>("LOGISTIC", OperatorType::kLogistic); |
| 121 | CheckSimpleOperator<TanhOperator>("TANH", OperatorType::kTanh); |
| 122 | CheckSimpleOperator<ExpOperator>("EXP", OperatorType::kExp); |
| 123 | CheckSimpleOperator<CosOperator>("COS", OperatorType::kCos); |
| 124 | CheckSimpleOperator<LogSoftmaxOperator>("LOG_SOFTMAX", |
| 125 | OperatorType::kLogSoftmax); |
| 126 | CheckSimpleOperator<TensorFlowMaximumOperator>( |
| 127 | "MAXIMUM", OperatorType::kMaximum); // Element-wise Maximum |
| 128 | CheckSimpleOperator<TensorFlowMinimumOperator>( |
| 129 | "MINIMUM", OperatorType::kMinimum); // Element-wise Minimum |
| 130 | CheckSimpleOperator<TensorFlowLessOperator>("LESS", OperatorType::kLess); |
| 131 | CheckSimpleOperator<NegOperator>("NEG", OperatorType::kNeg); |
| 132 | CheckSimpleOperator<SelectOperator>("SELECT", OperatorType::kSelect); |
| 133 | CheckSimpleOperator<SliceOperator>("SLICE", OperatorType::kSlice); |
| 134 | CheckSimpleOperator<SinOperator>("SIN", OperatorType::kSin); |
| 135 | CheckSimpleOperator<TensorFlowEqualOperator>("EQUAL", OperatorType::kEqual); |
| 136 | CheckSimpleOperator<TensorFlowNotEqualOperator>("NOT_EQUAL", |
| 137 | OperatorType::kNotEqual); |
| 138 | CheckSimpleOperator<LogOperator>("LOG", OperatorType::kLog); |
| 139 | CheckSimpleOperator<TensorFlowSqrtOperator>("SQRT", OperatorType::kSqrt); |
| 140 | CheckSimpleOperator<TensorFlowRsqrtOperator>("RSQRT", OperatorType::kRsqrt); |
| 141 | CheckSimpleOperator<PowOperator>("POW", OperatorType::kPow); |
| 142 | CheckSimpleOperator<LogicalOrOperator>("LOGICAL_OR", |
| 143 | OperatorType::kLogicalOr); |
| 144 | CheckSimpleOperator<LogicalAndOperator>("LOGICAL_AND", |
| 145 | OperatorType::kLogicalAnd); |
| 146 | CheckSimpleOperator<LogicalNotOperator>("LOGICAL_NOT", |
| 147 | OperatorType::kLogicalNot); |
| 148 | CheckSimpleOperator<FloorDivOperator>("FLOOR_DIV", OperatorType::kFloorDiv); |
| 149 | CheckSimpleOperator<TensorFlowSquareOperator>("SQUARE", |
| 150 | OperatorType::kSquare); |
| 151 | CheckSimpleOperator<TensorFlowZerosLikeOperator>("ZEROS_LIKE", |
| 152 | OperatorType::kZerosLike); |
| 153 | CheckSimpleOperator<FloorModOperator>("FLOOR_MOD", OperatorType::kFloorMod); |
| 154 | CheckSimpleOperator<RangeOperator>("RANGE", OperatorType::kRange); |
| 155 | CheckSimpleOperator<FillOperator>("FILL", OperatorType::kFill); |
| 156 | CheckSimpleOperator<ReverseV2Operator>("REVERSE_V2", |
| 157 | OperatorType::kReverseV2); |
| 158 | CheckSimpleOperator<TensorFlowRankOperator>("RANK", OperatorType::kRank); |
| 159 | } |
| 160 | |
| 161 | TEST_F(OperatorTest, BuiltinAdd) { |
| 162 | AddOperator op; |
nothing calls this directly
no test coverage detected