@@ -2583,4 +2583,135 @@ resnet_50-feature_vector:
2583
2583
- Identity:0
2584
2584
rtol : 0.05
2585
2585
atol : 0.0005
2586
- concrete_function : 0
2586
+ concrete_function : 0
2587
+
2588
+ albert_base :
2589
+ disabled : true
2590
+ model : " /tmp/albert_base"
2591
+ model_type : saved_model
2592
+ input_get : get_random
2593
+ inputs :
2594
+ " input_mask:0 " : np.ones([1, 384], dtype=np.int32)
2595
+ " segment_ids:0 " : np.ones([1, 384], dtype=np.int32)
2596
+ " input_ids:0 " : np.ones([1, 384], dtype=np.int32)
2597
+ outputs :
2598
+ - bert/pooler/dense/Tanh:0
2599
+ - bert/encoder/transformer/group_0_11/layer_11/inner_group_0/LayerNorm_1/batchnorm/add_1:0
2600
+ tag : " "
2601
+ signature_def : " tokens"
2602
+ rtol : 0.5
2603
+
2604
+ albert_lite :
2605
+ disabled : true
2606
+ model : " /tmp/albert-lite"
2607
+ model_type : saved_model
2608
+ input_get : get_random
2609
+ inputs :
2610
+ " input_mask:0 " : np.ones([1, 384], dtype=np.int32)
2611
+ " segment_ids:0 " : np.ones([1, 384], dtype=np.int32)
2612
+ " input_ids:0 " : np.ones([1, 384], dtype=np.int32)
2613
+ outputs :
2614
+ - bert/pooler/dense/Tanh:0
2615
+ - bert/encoder/transformer/group_0_11/layer_11/inner_group_0/LayerNorm_1/batchnorm/add_1:0
2616
+ tag : " "
2617
+ signature_def : " tokens"
2618
+ rtol : 0.5
2619
+
2620
+ albert_large :
2621
+ disabled : true
2622
+ model : " /tmp/albert_large"
2623
+ model_type : saved_model
2624
+ input_get : get_random
2625
+ inputs :
2626
+ " input_mask:0 " : np.ones([1, 384], dtype=np.int32)
2627
+ " segment_ids:0 " : np.ones([1, 384], dtype=np.int32)
2628
+ " input_ids:0 " : np.ones([1, 384], dtype=np.int32)
2629
+ outputs :
2630
+ - bert/pooler/dense/Tanh:0
2631
+ - bert/encoder/transformer/group_0_11/layer_11/inner_group_0/LayerNorm_1/batchnorm/add_1:0
2632
+ tag : " "
2633
+ signature_def : " tokens"
2634
+ rtol : 0.5
2635
+
2636
+ albert_xlarge :
2637
+ disabled : true
2638
+ model : " /tmp/albert_xlarge"
2639
+ model_type : saved_model
2640
+ input_get : get_random
2641
+ inputs :
2642
+ " input_mask:0 " : np.ones([1, 384], dtype=np.int32)
2643
+ " segment_ids:0 " : np.ones([1, 384], dtype=np.int32)
2644
+ " input_ids:0 " : np.ones([1, 384], dtype=np.int32)
2645
+ outputs :
2646
+ - bert/pooler/dense/Tanh:0
2647
+ - bert/encoder/transformer/group_0_11/layer_11/inner_group_0/LayerNorm_1/batchnorm/add_1:0
2648
+ tag : " "
2649
+ signature_def : " tokens"
2650
+ large_model : true
2651
+ check_only_shape : true
2652
+ structured_outputs :
2653
+ - pooled_output
2654
+ - sequence_output
2655
+ rtol : 0.5
2656
+
2657
+ albert_xxlarge :
2658
+ disabled : true
2659
+ model : " /tmp/albert_xxlarge"
2660
+ model_type : saved_model
2661
+ input_get : get_random
2662
+ inputs :
2663
+ " input_mask:0 " : np.ones([1, 384], dtype=np.int32)
2664
+ " segment_ids:0 " : np.ones([1, 384], dtype=np.int32)
2665
+ " input_ids:0 " : np.ones([1, 384], dtype=np.int32)
2666
+ outputs :
2667
+ - bert/pooler/dense/Tanh:0
2668
+ - bert/encoder/transformer/group_0_11/layer_11/inner_group_0/LayerNorm_1/batchnorm/add_1:0
2669
+ tag : " "
2670
+ signature_def : " tokens"
2671
+ large_model : true
2672
+ check_only_shape : true
2673
+ structured_outputs :
2674
+ - pooled_output
2675
+ - sequence_output
2676
+ rtol : 0.5
2677
+
2678
+ efficientnet-edgetpu-L :
2679
+ disabled : true
2680
+ url : https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/efficientnet-edgetpu-L.tar.gz
2681
+ model : efficientnet-edgetpu-L/saved_model
2682
+ model_type : saved_model
2683
+ input_get : get_beach
2684
+ inputs :
2685
+ " images:0 " : [1, 300, 300, 3]
2686
+ outputs :
2687
+ - Softmax:0
2688
+
2689
+ efficientnet-lite4 :
2690
+ disabled : true
2691
+ model : /data/NN/mlperf/efficientnet/efficientnet-lite4
2692
+ model_type : saved_model
2693
+ input_get : get_beach
2694
+ inputs :
2695
+ " images:0 " : [1, 224, 224, 3]
2696
+ outputs :
2697
+ - Softmax:0
2698
+
2699
+ efficientnet-qobvel-b0 :
2700
+ disabled : true
2701
+ model : /data/NN/mlperf/efficientnet/qobvel-b0
2702
+ model_type : saved_model
2703
+ input_get : get_beach
2704
+ inputs :
2705
+ " input_1:0 " : [1, 224, 224, 3]
2706
+ outputs :
2707
+ - Identity:0
2708
+
2709
+ efficientnet-qobvel-b1 :
2710
+ disabled : true
2711
+ model : /data/NN/mlperf/efficientnet/qobvel-b1
2712
+ model_type : saved_model
2713
+ input_get : get_beach
2714
+ inputs :
2715
+ " input_1:0 " : [1, 240, 240, 3]
2716
+ outputs :
2717
+ - Identity:0
0 commit comments