Build a vector containing all the known operators.
| 2425 | namespace { |
| 2426 | // Build a vector containing all the known operators. |
| 2427 | std::vector<std::unique_ptr<BaseOperator>> BuildOperatorList( |
| 2428 | bool enable_select_tf_ops = false) { |
| 2429 | std::vector<std::unique_ptr<BaseOperator>> ops; |
| 2430 | using tensorflow::MakeUnique; |
| 2431 | // Builtin Operators. |
| 2432 | ops.push_back( |
| 2433 | MakeUnique<Add>(::tflite::BuiltinOperator_ADD, OperatorType::kAdd)); |
| 2434 | ops.push_back( |
| 2435 | MakeUnique<AddN>(::tflite::BuiltinOperator_ADD_N, OperatorType::kAddN)); |
| 2436 | ops.push_back( |
| 2437 | MakeUnique<Div>(::tflite::BuiltinOperator_DIV, OperatorType::kDiv)); |
| 2438 | ops.push_back( |
| 2439 | MakeUnique<Sub>(::tflite::BuiltinOperator_SUB, OperatorType::kSub)); |
| 2440 | ops.push_back(MakeUnique<AveragePool>( |
| 2441 | ::tflite::BuiltinOperator_AVERAGE_POOL_2D, OperatorType::kAveragePool)); |
| 2442 | ops.push_back( |
| 2443 | MakeUnique<SpaceToBatchND>(::tflite::BuiltinOperator_SPACE_TO_BATCH_ND, |
| 2444 | OperatorType::kSpaceToBatchND)); |
| 2445 | ops.push_back( |
| 2446 | MakeUnique<BatchToSpaceND>(::tflite::BuiltinOperator_BATCH_TO_SPACE_ND, |
| 2447 | OperatorType::kBatchToSpaceND)); |
| 2448 | ops.push_back(MakeUnique<Concatenation>( |
| 2449 | ::tflite::BuiltinOperator_CONCATENATION, OperatorType::kConcatenation)); |
| 2450 | ops.push_back(MakeUnique<Convolution>(::tflite::BuiltinOperator_CONV_2D, |
| 2451 | OperatorType::kConv)); |
| 2452 | ops.push_back(MakeUnique<DepthwiseConvolution>( |
| 2453 | ::tflite::BuiltinOperator_DEPTHWISE_CONV_2D, |
| 2454 | OperatorType::kDepthwiseConv)); |
| 2455 | ops.push_back(MakeUnique<Dequantize>(::tflite::BuiltinOperator_DEQUANTIZE, |
| 2456 | OperatorType::kDequantize)); |
| 2457 | ops.push_back( |
| 2458 | MakeUnique<FullyConnected>(::tflite::BuiltinOperator_FULLY_CONNECTED, |
| 2459 | OperatorType::kFullyConnected)); |
| 2460 | ops.push_back(MakeUnique<Gather>(::tflite::BuiltinOperator_GATHER, |
| 2461 | OperatorType::kGather)); |
| 2462 | ops.push_back(MakeUnique<GatherNd>(::tflite::BuiltinOperator_GATHER_ND, |
| 2463 | OperatorType::kGatherNd)); |
| 2464 | ops.push_back( |
| 2465 | MakeUnique<L2Normalization>(::tflite::BuiltinOperator_L2_NORMALIZATION, |
| 2466 | OperatorType::kL2Normalization)); |
| 2467 | ops.push_back(MakeUnique<L2Pool>(::tflite::BuiltinOperator_L2_POOL_2D, |
| 2468 | OperatorType::kL2Pool)); |
| 2469 | ops.push_back(MakeUnique<LocalResponseNormalization>( |
| 2470 | ::tflite::BuiltinOperator_LOCAL_RESPONSE_NORMALIZATION, |
| 2471 | OperatorType::kLocalResponseNormalization)); |
| 2472 | ops.push_back(MakeUnique<MaxPool>(::tflite::BuiltinOperator_MAX_POOL_2D, |
| 2473 | OperatorType::kMaxPool)); |
| 2474 | ops.push_back( |
| 2475 | MakeUnique<Mul>(::tflite::BuiltinOperator_MUL, OperatorType::kMul)); |
| 2476 | |
| 2477 | ops.push_back( |
| 2478 | MakeUnique<Pad>(::tflite::BuiltinOperator_PAD, OperatorType::kPad)); |
| 2479 | ops.push_back( |
| 2480 | MakeUnique<PadV2>(::tflite::BuiltinOperator_PADV2, OperatorType::kPadV2)); |
| 2481 | ops.push_back(MakeUnique<Reshape>(::tflite::BuiltinOperator_RESHAPE, |
| 2482 | OperatorType::kReshape)); |
| 2483 | ops.push_back(MakeUnique<Softmax>(::tflite::BuiltinOperator_SOFTMAX, |
| 2484 | OperatorType::kSoftmax)); |
no test coverage detected