| 165 | } |
| 166 | |
| 167 | BuiltinRefOpResolver::BuiltinRefOpResolver() { |
| 168 | AddBuiltin(BuiltinOperator_ABS, Register_ABS()); |
| 169 | AddBuiltin(BuiltinOperator_RELU, Register_RELU()); |
| 170 | AddBuiltin(BuiltinOperator_RELU_N1_TO_1, Register_RELU_N1_TO_1()); |
| 171 | AddBuiltin(BuiltinOperator_RELU6, Register_RELU6()); |
| 172 | AddBuiltin(BuiltinOperator_TANH, Register_TANH_REF()); |
| 173 | AddBuiltin(BuiltinOperator_LOGISTIC, Register_LOGISTIC_REF()); |
| 174 | AddBuiltin(BuiltinOperator_AVERAGE_POOL_2D, Register_AVERAGE_POOL_REF()); |
| 175 | AddBuiltin(BuiltinOperator_MAX_POOL_2D, Register_MAX_POOL_REF()); |
| 176 | AddBuiltin(BuiltinOperator_L2_POOL_2D, Register_L2_POOL_REF()); |
| 177 | AddBuiltin(BuiltinOperator_CONV_2D, Register_CONVOLUTION_REF()); |
| 178 | AddBuiltin(BuiltinOperator_DEPTHWISE_CONV_2D, |
| 179 | Register_DEPTHWISE_CONVOLUTION_REF(), |
| 180 | /* min_version */ 1, |
| 181 | /* max_version */ 2); |
| 182 | AddBuiltin(BuiltinOperator_SVDF, Register_SVDF()); |
| 183 | AddBuiltin(BuiltinOperator_RNN, Register_RNN()); |
| 184 | AddBuiltin(BuiltinOperator_BIDIRECTIONAL_SEQUENCE_RNN, |
| 185 | Register_BIDIRECTIONAL_SEQUENCE_RNN()); |
| 186 | AddBuiltin(BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_RNN, |
| 187 | Register_UNIDIRECTIONAL_SEQUENCE_RNN()); |
| 188 | AddBuiltin(BuiltinOperator_EMBEDDING_LOOKUP, Register_EMBEDDING_LOOKUP()); |
| 189 | AddBuiltin(BuiltinOperator_EMBEDDING_LOOKUP_SPARSE, |
| 190 | Register_EMBEDDING_LOOKUP_SPARSE()); |
| 191 | AddBuiltin(BuiltinOperator_FULLY_CONNECTED, Register_FULLY_CONNECTED_REF(), |
| 192 | /* min_version */ 1, |
| 193 | /* max_version */ 2); |
| 194 | AddBuiltin(BuiltinOperator_LSH_PROJECTION, Register_LSH_PROJECTION()); |
| 195 | AddBuiltin(BuiltinOperator_HASHTABLE_LOOKUP, Register_HASHTABLE_LOOKUP()); |
| 196 | AddBuiltin(BuiltinOperator_SOFTMAX, Register_SOFTMAX()); |
| 197 | AddBuiltin(BuiltinOperator_CONCATENATION, Register_CONCATENATION_REF()); |
| 198 | AddBuiltin(BuiltinOperator_ADD, Register_ADD_REF()); |
| 199 | AddBuiltin(BuiltinOperator_SPACE_TO_BATCH_ND, |
| 200 | Register_SPACE_TO_BATCH_ND_REF()); |
| 201 | AddBuiltin(BuiltinOperator_BATCH_TO_SPACE_ND, |
| 202 | Register_BATCH_TO_SPACE_ND_REF()); |
| 203 | AddBuiltin(BuiltinOperator_MUL, Register_MUL_REF()); |
| 204 | AddBuiltin(BuiltinOperator_L2_NORMALIZATION, Register_L2NORM_REF()); |
| 205 | AddBuiltin(BuiltinOperator_LOCAL_RESPONSE_NORMALIZATION, |
| 206 | Register_LOCAL_RESPONSE_NORM_REF()); |
| 207 | AddBuiltin(BuiltinOperator_LSTM, Register_LSTM(), /* min_version */ 1, |
| 208 | /* max_version */ 2); |
| 209 | AddBuiltin(BuiltinOperator_BIDIRECTIONAL_SEQUENCE_LSTM, |
| 210 | Register_BIDIRECTIONAL_SEQUENCE_LSTM(), /* min_version */ 1, |
| 211 | /* max_version */ 2); |
| 212 | AddBuiltin(BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM, |
| 213 | Register_UNIDIRECTIONAL_SEQUENCE_LSTM()); |
| 214 | AddBuiltin(BuiltinOperator_PAD, Register_PAD_REF()); |
| 215 | AddBuiltin(BuiltinOperator_PADV2, Register_PADV2_REF()); |
| 216 | AddBuiltin(BuiltinOperator_RESHAPE, Register_RESHAPE()); |
| 217 | AddBuiltin(BuiltinOperator_RESIZE_BILINEAR, Register_RESIZE_BILINEAR_REF()); |
| 218 | AddBuiltin(BuiltinOperator_RESIZE_NEAREST_NEIGHBOR, |
| 219 | Register_RESIZE_NEAREST_NEIGHBOR_REF()); |
| 220 | AddBuiltin(BuiltinOperator_SKIP_GRAM, Register_SKIP_GRAM()); |
| 221 | AddBuiltin(BuiltinOperator_SPACE_TO_DEPTH, Register_SPACE_TO_DEPTH_REF()); |
| 222 | AddBuiltin(BuiltinOperator_DEPTH_TO_SPACE, Register_DEPTH_TO_SPACE_REF()); |
| 223 | AddBuiltin(BuiltinOperator_GATHER, Register_GATHER()); |
| 224 | AddBuiltin(BuiltinOperator_TRANSPOSE, Register_TRANSPOSE_REF()); |
nothing calls this directly
no test coverage detected