Ejemplo n.º 1
0
    def test_decode_name_with_port_delim_equal_data_in(self):
        set_equal_data('conv2d', 'conv2d')
        node = decode_name_with_port(self.model, '0:conv2d')
        model_stat = get_model_statistic()
        place_stat = get_place_statistic()

        assert model_stat.get_place_by_tensor_name == 1
        assert model_stat.get_place_by_operation_name == 2
        assert place_stat.get_input_port == 1
        assert place_stat.is_equal_data > 0
        assert node
Ejemplo n.º 2
0
    def test_decode_name_with_port_delim_all_same_data(self):
        set_equal_data('8', '9')
        node = decode_name_with_port(self.model, '8:9')
        model_stat = get_model_statistic()
        place_stat = get_place_statistic()

        assert model_stat.get_place_by_tensor_name == 1
        assert model_stat.get_place_by_operation_name == 3
        assert place_stat.get_input_port == 1
        assert place_stat.get_output_port == 1
        # At least 3 comparisons of places are expected
        assert place_stat.is_equal_data > 2
        assert node