980904c9ec
* Moved and merged mo/ and extensions/ into openvino/tools/mo
* edited imports
* edited docs to use mo script from entry_point
* edited MO transformations list loading and setup.py
* changed full path -> 'mo' entry point in docs (leftovers)
* corrected package_BOM
* updated resolving --transformation_config in cli_parser.py
* pkgutil-style __init__.py, added summarize_graph into entry points
* updated DOCs for the new --transformations_config
* fix select
* updated install instructions, fixed setup.py for windows and python_version < 3.8
* fixed typo in requirements.txt
* resolved conflicts
* removed creating custom __init__.py from setup.py
* corrected folder with caffe proto
* corrected loading user defined extensions
* fix openvino.tools.mo import in serialize.py
* corrected layer tests for new namespace
* fix in get_testdata.py
* moved model-optimizer into tools/
* renamed import in POT
* corrected mo.yml
* correct CMakeLists.txt for the newest tools/mo
* corrected find_ie_version.py
* docs and openvino-dev setup.py update for the newest tools/mo
* miscellaneous leftovers and fixes
* corrected CI files, pybind11_add_module in CMakeLists.txt and use of tools/mo path instead of tools/model_optimizer
* add_subdirectory pybind11 for tools/mo
* POT path fix
* updated setupvars.sh setupvars.bat
* Revert "updated setupvars.sh setupvars.bat"
This reverts commit c011142340.
* removed model-optimizer env variables from setupvars
* updated CMakeLists.txt to pack MO properly with tests component
* corrected left imports, corrected loading requirements for layer tests
* mo doc typo correction
* minor corrections in docs; removed summarize_graph from entry_points
* get_started_windows.md, MonoDepth_how_to.md corrections, mo path corrections
139 lines
6.5 KiB
Python
139 lines
6.5 KiB
Python
# Copyright (C) 2018-2021 Intel Corporation
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from distutils.version import LooseVersion
|
|
|
|
import numpy as np
|
|
import pytest
|
|
from common.layer_test_class import check_ir_version
|
|
from common.tf_layer_test_class import CommonTFLayerTest
|
|
from common.utils.tf_utils import permute_nchw_to_nhwc
|
|
from openvino.tools.mo.front.common.partial_infer.utils import int64_array
|
|
from unit_tests.utils.graph import build_graph
|
|
|
|
|
|
class TestLogSoftmax(CommonTFLayerTest):
|
|
def create_log_softmax_net(self, shape, reduction_axis, ir_version, use_new_frontend):
|
|
"""
|
|
Tensorflow net IR net
|
|
|
|
Input->LogSoftmax => Input->Softmax->Log
|
|
|
|
"""
|
|
|
|
#
|
|
# Create Tensorflow model
|
|
#
|
|
|
|
import tensorflow as tf
|
|
|
|
tf.compat.v1.reset_default_graph()
|
|
|
|
# Create the graph and model
|
|
with tf.compat.v1.Session() as sess:
|
|
tf_x_shape = shape.copy()
|
|
|
|
tf_x_shape = permute_nchw_to_nhwc(tf_x_shape, use_new_frontend)
|
|
input = tf.compat.v1.placeholder(tf.float32, tf_x_shape, 'Input')
|
|
if LooseVersion(tf.__version__) < LooseVersion('2.0.0'):
|
|
tf.nn.log_softmax(input, name='Operation', axis=reduction_axis)
|
|
else:
|
|
tf.nn.log_softmax(input, axis=reduction_axis, name='Operation')
|
|
|
|
tf.compat.v1.global_variables_initializer()
|
|
tf_net = sess.graph_def
|
|
|
|
ref_net = None
|
|
|
|
reduce_sum_shape = np.copy(shape)
|
|
rank = len(shape)
|
|
if rank in {4, 5}:
|
|
reduction_axis = reduction_axis if reduction_axis >= 0 else rank + reduction_axis
|
|
if rank == 4:
|
|
reduction_axis = {0: 0, 1: 2, 2: 3, 3: 1}[reduction_axis]
|
|
else:
|
|
reduction_axis = {0: 0, 1: 2, 2: 3, 3: 4, 4: 1}[reduction_axis]
|
|
|
|
reduce_sum_shape[reduction_axis] = 1
|
|
|
|
converted_shape = shape if rank != 1 else shape[0]
|
|
if check_ir_version(10, None, ir_version) and not use_new_frontend:
|
|
ref_nodes_attributes = {
|
|
'input': {'kind': 'op', 'type': 'Parameter', 'shape': converted_shape},
|
|
'input_data': {'shape': shape, 'kind': 'data', 'value': None},
|
|
'reduce_max_axis_val': {'shape': int64_array([reduction_axis]).shape,
|
|
'kind': 'data',
|
|
'value': int64_array([reduction_axis])},
|
|
'reduce_max_axis': {'type': 'Const', 'kind': 'op', 'shape': 1},
|
|
'reduce_max_axis_data': {'shape': int64_array([1]), 'kind': 'data', 'value': None},
|
|
'reduce_max': {'type': 'ReduceMax', 'kind': 'op', 'keep_dims': True},
|
|
'reduce_max_data': {'shape': reduce_sum_shape, 'kind': 'data', 'value': None},
|
|
'sub_first': {'type': 'Subtract', 'kind': 'op'},
|
|
'sub_first_data': {'shape': shape, 'kind': 'data', 'value': None},
|
|
'reduce_sum_axis_val': {'shape': int64_array([reduction_axis]).shape,
|
|
'kind': 'data',
|
|
'value': int64_array([reduction_axis])},
|
|
'reduce_sum_axis': {'type': 'Const', 'kind': 'op', 'shape': 1},
|
|
'reduce_sum_axis_data': {'shape': int64_array([1]), 'kind': 'data', 'value': None},
|
|
'reduce_sum': {'type': 'ReduceSum', 'kind': 'op', 'keep_dims': True},
|
|
'reduce_sum_data': {'shape': reduce_sum_shape, 'kind': 'data', 'value': None},
|
|
'exp': {'type': 'Exp', 'kind': 'op'},
|
|
'exp_data': {'shape': shape, 'kind': 'data', 'value': None},
|
|
'log': {'type': 'Log', 'kind': 'op'},
|
|
'log_data': {'shape': reduce_sum_shape, 'kind': 'data', 'value': None},
|
|
'sub_second': {'type': 'Subtract', 'kind': 'op'},
|
|
'sub_second_data': {'shape': shape, 'kind': 'data', 'value': None},
|
|
'result': {'kind': 'op', 'type': 'Result'},
|
|
}
|
|
|
|
ref_edges = [
|
|
('input', 'input_data'),
|
|
('reduce_max_axis_val', 'reduce_max_axis'),
|
|
('reduce_max_axis', 'reduce_max_axis_data'),
|
|
('reduce_max_axis_data', 'reduce_max', {'in': 1}),
|
|
('reduce_max', 'reduce_max_data'),
|
|
('input_data', 'reduce_max', {'out': 0, 'in': 0}),
|
|
('input_data', 'sub_first', {'out': 0, 'in': 0}),
|
|
('reduce_max_data', 'sub_first', {'in': 1}),
|
|
('sub_first', 'sub_first_data'),
|
|
('reduce_sum_axis_val', 'reduce_sum_axis'),
|
|
('reduce_sum_axis', 'reduce_sum_axis_data'),
|
|
('reduce_sum_axis_data', 'reduce_sum', {'in': 1}),
|
|
('reduce_sum', 'reduce_sum_data'),
|
|
('sub_first_data', 'exp'),
|
|
('exp', 'exp_data'),
|
|
('exp_data', 'reduce_sum', {'in': 0}),
|
|
('reduce_sum_data', 'log'),
|
|
('log', 'log_data'),
|
|
('log_data', 'sub_second', {'in': 1}),
|
|
('sub_second', 'sub_second_data'),
|
|
('sub_first_data', 'sub_second', {'out': 0, 'in': 0}),
|
|
('sub_second_data', 'result'),
|
|
]
|
|
|
|
ref_net = build_graph(ref_nodes_attributes, ref_edges)
|
|
|
|
return tf_net, ref_net
|
|
|
|
test_data_precommit = [
|
|
pytest.param(dict(shape=[3, 2, 3, 7, 6], reduction_axis=-1),
|
|
marks=pytest.mark.skip(reason="Skipped until fixed"))
|
|
]
|
|
|
|
@pytest.mark.parametrize("params", test_data_precommit)
|
|
@pytest.mark.precommit
|
|
def test_log_softmax_precommit(self, params, ie_device, precision, ir_version, temp_dir, use_new_frontend):
|
|
self._test(*self.create_log_softmax_net(**params, ir_version=ir_version, use_new_frontend=use_new_frontend),
|
|
ie_device, precision, ir_version, temp_dir=temp_dir, use_new_frontend=use_new_frontend)
|
|
|
|
test_data = [dict(shape=[1], reduction_axis=-1),
|
|
dict(shape=[2, 5], reduction_axis=-1),
|
|
dict(shape=[5, 3, 7, 4], reduction_axis=-1),
|
|
dict(shape=[3, 2, 3, 7, 6], reduction_axis=-1)]
|
|
|
|
@pytest.mark.parametrize("params", test_data)
|
|
@pytest.mark.nightly
|
|
def test_log_softmax(self, params, ie_device, precision, ir_version, temp_dir, use_new_frontend):
|
|
self._test(*self.create_log_softmax_net(**params, ir_version=ir_version, use_new_frontend=use_new_frontend),
|
|
ie_device, precision, ir_version, temp_dir=temp_dir, use_new_frontend=use_new_frontend)
|