add f64 support to ngraph serializer (#3462)
* add f64 support to ngraph serializer * Create add_abc test with f64 type
This commit is contained in:
Piotr Szmelczynski
GitHub
parent
291785dc14
commit
3d66869081
@ -240,6 +240,8 @@ std::string get_output_precision_name(ngraph::Output<Node>& o) {
|
|||||||
return "FP32";
|
return "FP32";
|
||||||
case ::ngraph::element::Type_t::bf16:
|
case ::ngraph::element::Type_t::bf16:
|
||||||
return "BF16";
|
return "BF16";
|
||||||
|
case ::ngraph::element::Type_t::f64:
|
||||||
|
return "FP64";
|
||||||
case ::ngraph::element::Type_t::i8:
|
case ::ngraph::element::Type_t::i8:
|
||||||
return "I8";
|
return "I8";
|
||||||
case ::ngraph::element::Type_t::i16:
|
case ::ngraph::element::Type_t::i16:
|
||||||
|
|||||||
@ -0,0 +1,112 @@
|
|||||||
|
<?xml version="1.0" ?>
|
||||||
|
<net name="add_abc" version="10">
|
||||||
|
<layers>
|
||||||
|
<layer id="0" name="A" type="Parameter" version="opset1">
|
||||||
|
<data element_type="f64" shape="1"/>
|
||||||
|
<output>
|
||||||
|
<port id="0" precision="FP64">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</output>
|
||||||
|
</layer>
|
||||||
|
<layer id="1" name="B" type="Parameter" version="opset1">
|
||||||
|
<data element_type="f64" shape="1"/>
|
||||||
|
<output>
|
||||||
|
<port id="0" precision="FP64">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</output>
|
||||||
|
</layer>
|
||||||
|
<layer id="2" name="add_node1" type="Add" version="opset1">
|
||||||
|
<input>
|
||||||
|
<port id="0">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
<port id="1">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</input>
|
||||||
|
<output>
|
||||||
|
<port id="2" precision="FP64">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</output>
|
||||||
|
</layer>
|
||||||
|
<layer id="3" name="C" type="Parameter" version="opset1">
|
||||||
|
<data element_type="f64" shape="1"/>
|
||||||
|
<output>
|
||||||
|
<port id="0" precision="FP64">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</output>
|
||||||
|
</layer>
|
||||||
|
<layer id="4" name="Y" type="Add" version="opset1">
|
||||||
|
<input>
|
||||||
|
<port id="0">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
<port id="1">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</input>
|
||||||
|
<output>
|
||||||
|
<port id="2" precision="FP64">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</output>
|
||||||
|
</layer>
|
||||||
|
<layer id="5" name="Y/sink_port_0" type="Result" version="opset1">
|
||||||
|
<input>
|
||||||
|
<port id="0">
|
||||||
|
<dim>1</dim>
|
||||||
|
</port>
|
||||||
|
</input>
|
||||||
|
</layer>
|
||||||
|
</layers>
|
||||||
|
<edges>
|
||||||
|
<edge from-layer="0" from-port="0" to-layer="2" to-port="0"/>
|
||||||
|
<edge from-layer="1" from-port="0" to-layer="2" to-port="1"/>
|
||||||
|
<edge from-layer="2" from-port="2" to-layer="4" to-port="0"/>
|
||||||
|
<edge from-layer="3" from-port="0" to-layer="4" to-port="1"/>
|
||||||
|
<edge from-layer="4" from-port="2" to-layer="5" to-port="0"/>
|
||||||
|
</edges>
|
||||||
|
<meta_data>
|
||||||
|
<MO_version value="unknown version"/>
|
||||||
|
<cli_parameters>
|
||||||
|
<caffe_parser_path value="DIR"/>
|
||||||
|
<data_type value="float"/>
|
||||||
|
<disable_nhwc_to_nchw value="False"/>
|
||||||
|
<disable_omitting_optional value="False"/>
|
||||||
|
<disable_resnet_optimization value="False"/>
|
||||||
|
<disable_weights_compression value="False"/>
|
||||||
|
<enable_concat_optimization value="False"/>
|
||||||
|
<enable_flattening_nested_params value="False"/>
|
||||||
|
<enable_ssd_gluoncv value="False"/>
|
||||||
|
<extensions value="DIR"/>
|
||||||
|
<framework value="onnx"/>
|
||||||
|
<freeze_placeholder_with_value value="{}"/>
|
||||||
|
<generate_deprecated_IR_V7 value="False"/>
|
||||||
|
<input_model value="DIR/add_abc.onnx"/>
|
||||||
|
<input_model_is_text value="False"/>
|
||||||
|
<k value="DIR/CustomLayersMapping.xml"/>
|
||||||
|
<keep_shape_ops value="True"/>
|
||||||
|
<legacy_mxnet_model value="False"/>
|
||||||
|
<log_level value="ERROR"/>
|
||||||
|
<mean_scale_values value="{}"/>
|
||||||
|
<mean_values value="()"/>
|
||||||
|
<model_name value="add_abc"/>
|
||||||
|
<output_dir value="DIR"/>
|
||||||
|
<placeholder_data_types value="{}"/>
|
||||||
|
<progress value="False"/>
|
||||||
|
<remove_memory value="False"/>
|
||||||
|
<remove_output_softmax value="False"/>
|
||||||
|
<reverse_input_channels value="False"/>
|
||||||
|
<save_params_from_nd value="False"/>
|
||||||
|
<scale_values value="()"/>
|
||||||
|
<silent value="False"/>
|
||||||
|
<static_shape value="False"/>
|
||||||
|
<stream_output value="False"/>
|
||||||
|
<unset unset_cli_parameters="batch, counts, disable_fusing, disable_gfusing, finegrain_fusing, input, input_checkpoint, input_meta_graph, input_proto, input_shape, input_symbol, mean_file, mean_file_offsets, move_to_preprocess, nd_prefix_name, output, placeholder_shapes, pretrained_model_name, saved_model_dir, saved_model_tags, scale, tensorboard_logdir, tensorflow_custom_layer_libraries, tensorflow_custom_operations_config_update, tensorflow_object_detection_api_pipeline_config, tensorflow_use_custom_operations_config, transformations_config"/>
|
||||||
|
</cli_parameters>
|
||||||
|
</meta_data>
|
||||||
|
</net>
|
||||||
@ -49,6 +49,7 @@ TEST_P(SerializationTest, CompareFunctions) {
|
|||||||
|
|
||||||
INSTANTIATE_TEST_CASE_P(IRSerialization, SerializationTest,
|
INSTANTIATE_TEST_CASE_P(IRSerialization, SerializationTest,
|
||||||
testing::Values(std::make_tuple("add_abc.xml"),
|
testing::Values(std::make_tuple("add_abc.xml"),
|
||||||
|
std::make_tuple("add_abc_f64.xml"),
|
||||||
std::make_tuple("split_equal_parts_2d.xml"),
|
std::make_tuple("split_equal_parts_2d.xml"),
|
||||||
std::make_tuple("addmul_abc.xml"),
|
std::make_tuple("addmul_abc.xml"),
|
||||||
std::make_tuple("add_abc_initializers.xml"),
|
std::make_tuple("add_abc_initializers.xml"),
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user