Update Python API to use opset8 (#6535)
This commit is contained in:
parent
72a4d6d9d9
commit
b64e146a90
@ -28,159 +28,159 @@ from ngraph.frontend import OpValidationFailure
|
||||
from ngraph.frontend import Place
|
||||
from ngraph.helpers import function_from_cnn
|
||||
from ngraph.helpers import function_to_cnn
|
||||
from ngraph.opset7 import absolute
|
||||
from ngraph.opset7 import absolute as abs
|
||||
from ngraph.opset7 import acos
|
||||
from ngraph.opset7 import acosh
|
||||
from ngraph.opset7 import add
|
||||
from ngraph.opset7 import asin
|
||||
from ngraph.opset7 import asinh
|
||||
from ngraph.opset7 import assign
|
||||
from ngraph.opset7 import atan
|
||||
from ngraph.opset7 import atanh
|
||||
from ngraph.opset7 import avg_pool
|
||||
from ngraph.opset7 import batch_norm_inference
|
||||
from ngraph.opset7 import batch_to_space
|
||||
from ngraph.opset7 import binary_convolution
|
||||
from ngraph.opset7 import broadcast
|
||||
from ngraph.opset7 import bucketize
|
||||
from ngraph.opset7 import ceiling
|
||||
from ngraph.opset7 import ceiling as ceil
|
||||
from ngraph.opset7 import clamp
|
||||
from ngraph.opset7 import concat
|
||||
from ngraph.opset7 import constant
|
||||
from ngraph.opset7 import convert
|
||||
from ngraph.opset7 import convert_like
|
||||
from ngraph.opset7 import convolution
|
||||
from ngraph.opset7 import convolution_backprop_data
|
||||
from ngraph.opset7 import cos
|
||||
from ngraph.opset7 import cosh
|
||||
from ngraph.opset7 import ctc_greedy_decoder
|
||||
from ngraph.opset7 import ctc_greedy_decoder_seq_len
|
||||
from ngraph.opset7 import ctc_loss
|
||||
from ngraph.opset7 import cum_sum
|
||||
from ngraph.opset7 import cum_sum as cumsum
|
||||
from ngraph.opset7 import deformable_convolution
|
||||
from ngraph.opset7 import deformable_psroi_pooling
|
||||
from ngraph.opset7 import depth_to_space
|
||||
from ngraph.opset7 import detection_output
|
||||
from ngraph.opset7 import dft
|
||||
from ngraph.opset7 import divide
|
||||
from ngraph.opset7 import einsum
|
||||
from ngraph.opset7 import elu
|
||||
from ngraph.opset7 import embedding_bag_offsets_sum
|
||||
from ngraph.opset7 import embedding_bag_packed_sum
|
||||
from ngraph.opset7 import embedding_segments_sum
|
||||
from ngraph.opset7 import extract_image_patches
|
||||
from ngraph.opset7 import equal
|
||||
from ngraph.opset7 import erf
|
||||
from ngraph.opset7 import exp
|
||||
from ngraph.opset7 import fake_quantize
|
||||
from ngraph.opset7 import floor
|
||||
from ngraph.opset7 import floor_mod
|
||||
from ngraph.opset7 import gather
|
||||
from ngraph.opset7 import gather_elements
|
||||
from ngraph.opset7 import gather_nd
|
||||
from ngraph.opset7 import gather_tree
|
||||
from ngraph.opset7 import gelu
|
||||
from ngraph.opset7 import greater
|
||||
from ngraph.opset7 import greater_equal
|
||||
from ngraph.opset7 import grn
|
||||
from ngraph.opset7 import group_convolution
|
||||
from ngraph.opset7 import group_convolution_backprop_data
|
||||
from ngraph.opset7 import gru_cell
|
||||
from ngraph.opset7 import gru_sequence
|
||||
from ngraph.opset7 import hard_sigmoid
|
||||
from ngraph.opset7 import hsigmoid
|
||||
from ngraph.opset7 import hswish
|
||||
from ngraph.opset7 import idft
|
||||
from ngraph.opset7 import interpolate
|
||||
from ngraph.opset7 import less
|
||||
from ngraph.opset7 import less_equal
|
||||
from ngraph.opset7 import log
|
||||
from ngraph.opset7 import logical_and
|
||||
from ngraph.opset7 import logical_not
|
||||
from ngraph.opset7 import logical_or
|
||||
from ngraph.opset7 import logical_xor
|
||||
from ngraph.opset7 import log_softmax
|
||||
from ngraph.opset7 import loop
|
||||
from ngraph.opset7 import lrn
|
||||
from ngraph.opset7 import lstm_cell
|
||||
from ngraph.opset7 import lstm_sequence
|
||||
from ngraph.opset7 import matmul
|
||||
from ngraph.opset7 import max_pool
|
||||
from ngraph.opset7 import maximum
|
||||
from ngraph.opset7 import minimum
|
||||
from ngraph.opset7 import mish
|
||||
from ngraph.opset7 import mod
|
||||
from ngraph.opset7 import multiply
|
||||
from ngraph.opset7 import mvn
|
||||
from ngraph.opset7 import negative
|
||||
from ngraph.opset7 import non_max_suppression
|
||||
from ngraph.opset7 import non_zero
|
||||
from ngraph.opset7 import normalize_l2
|
||||
from ngraph.opset7 import not_equal
|
||||
from ngraph.opset7 import one_hot
|
||||
from ngraph.opset7 import pad
|
||||
from ngraph.opset7 import parameter
|
||||
from ngraph.opset7 import power
|
||||
from ngraph.opset7 import prelu
|
||||
from ngraph.opset7 import prior_box
|
||||
from ngraph.opset7 import prior_box_clustered
|
||||
from ngraph.opset7 import psroi_pooling
|
||||
from ngraph.opset7 import proposal
|
||||
from ngraph.opset7 import range
|
||||
from ngraph.opset7 import read_value
|
||||
from ngraph.opset7 import reduce_l1
|
||||
from ngraph.opset7 import reduce_l2
|
||||
from ngraph.opset7 import reduce_logical_and
|
||||
from ngraph.opset7 import reduce_logical_or
|
||||
from ngraph.opset7 import reduce_max
|
||||
from ngraph.opset7 import reduce_mean
|
||||
from ngraph.opset7 import reduce_min
|
||||
from ngraph.opset7 import reduce_prod
|
||||
from ngraph.opset7 import reduce_sum
|
||||
from ngraph.opset7 import region_yolo
|
||||
from ngraph.opset7 import reorg_yolo
|
||||
from ngraph.opset7 import relu
|
||||
from ngraph.opset7 import reshape
|
||||
from ngraph.opset7 import result
|
||||
from ngraph.opset7 import reverse_sequence
|
||||
from ngraph.opset7 import rnn_cell
|
||||
from ngraph.opset7 import rnn_sequence
|
||||
from ngraph.opset7 import roi_align
|
||||
from ngraph.opset7 import roi_pooling
|
||||
from ngraph.opset7 import roll
|
||||
from ngraph.opset7 import round
|
||||
from ngraph.opset7 import scatter_elements_update
|
||||
from ngraph.opset7 import scatter_update
|
||||
from ngraph.opset7 import select
|
||||
from ngraph.opset7 import selu
|
||||
from ngraph.opset7 import shape_of
|
||||
from ngraph.opset7 import shuffle_channels
|
||||
from ngraph.opset7 import sigmoid
|
||||
from ngraph.opset7 import sign
|
||||
from ngraph.opset7 import sin
|
||||
from ngraph.opset7 import sinh
|
||||
from ngraph.opset7 import softmax
|
||||
from ngraph.opset7 import softplus
|
||||
from ngraph.opset7 import space_to_batch
|
||||
from ngraph.opset7 import space_to_depth
|
||||
from ngraph.opset7 import split
|
||||
from ngraph.opset7 import sqrt
|
||||
from ngraph.opset7 import squared_difference
|
||||
from ngraph.opset7 import squeeze
|
||||
from ngraph.opset7 import strided_slice
|
||||
from ngraph.opset7 import subtract
|
||||
from ngraph.opset7 import swish
|
||||
from ngraph.opset7 import tan
|
||||
from ngraph.opset7 import tanh
|
||||
from ngraph.opset7 import tensor_iterator
|
||||
from ngraph.opset7 import tile
|
||||
from ngraph.opset7 import topk
|
||||
from ngraph.opset7 import transpose
|
||||
from ngraph.opset7 import unsqueeze
|
||||
from ngraph.opset7 import variadic_split
|
||||
from ngraph.opset8 import absolute
|
||||
from ngraph.opset8 import absolute as abs
|
||||
from ngraph.opset8 import acos
|
||||
from ngraph.opset8 import acosh
|
||||
from ngraph.opset8 import add
|
||||
from ngraph.opset8 import asin
|
||||
from ngraph.opset8 import asinh
|
||||
from ngraph.opset8 import assign
|
||||
from ngraph.opset8 import atan
|
||||
from ngraph.opset8 import atanh
|
||||
from ngraph.opset8 import avg_pool
|
||||
from ngraph.opset8 import batch_norm_inference
|
||||
from ngraph.opset8 import batch_to_space
|
||||
from ngraph.opset8 import binary_convolution
|
||||
from ngraph.opset8 import broadcast
|
||||
from ngraph.opset8 import bucketize
|
||||
from ngraph.opset8 import ceiling
|
||||
from ngraph.opset8 import ceiling as ceil
|
||||
from ngraph.opset8 import clamp
|
||||
from ngraph.opset8 import concat
|
||||
from ngraph.opset8 import constant
|
||||
from ngraph.opset8 import convert
|
||||
from ngraph.opset8 import convert_like
|
||||
from ngraph.opset8 import convolution
|
||||
from ngraph.opset8 import convolution_backprop_data
|
||||
from ngraph.opset8 import cos
|
||||
from ngraph.opset8 import cosh
|
||||
from ngraph.opset8 import ctc_greedy_decoder
|
||||
from ngraph.opset8 import ctc_greedy_decoder_seq_len
|
||||
from ngraph.opset8 import ctc_loss
|
||||
from ngraph.opset8 import cum_sum
|
||||
from ngraph.opset8 import cum_sum as cumsum
|
||||
from ngraph.opset8 import deformable_convolution
|
||||
from ngraph.opset8 import deformable_psroi_pooling
|
||||
from ngraph.opset8 import depth_to_space
|
||||
from ngraph.opset8 import detection_output
|
||||
from ngraph.opset8 import dft
|
||||
from ngraph.opset8 import divide
|
||||
from ngraph.opset8 import einsum
|
||||
from ngraph.opset8 import elu
|
||||
from ngraph.opset8 import embedding_bag_offsets_sum
|
||||
from ngraph.opset8 import embedding_bag_packed_sum
|
||||
from ngraph.opset8 import embedding_segments_sum
|
||||
from ngraph.opset8 import extract_image_patches
|
||||
from ngraph.opset8 import equal
|
||||
from ngraph.opset8 import erf
|
||||
from ngraph.opset8 import exp
|
||||
from ngraph.opset8 import fake_quantize
|
||||
from ngraph.opset8 import floor
|
||||
from ngraph.opset8 import floor_mod
|
||||
from ngraph.opset8 import gather
|
||||
from ngraph.opset8 import gather_elements
|
||||
from ngraph.opset8 import gather_nd
|
||||
from ngraph.opset8 import gather_tree
|
||||
from ngraph.opset8 import gelu
|
||||
from ngraph.opset8 import greater
|
||||
from ngraph.opset8 import greater_equal
|
||||
from ngraph.opset8 import grn
|
||||
from ngraph.opset8 import group_convolution
|
||||
from ngraph.opset8 import group_convolution_backprop_data
|
||||
from ngraph.opset8 import gru_cell
|
||||
from ngraph.opset8 import gru_sequence
|
||||
from ngraph.opset8 import hard_sigmoid
|
||||
from ngraph.opset8 import hsigmoid
|
||||
from ngraph.opset8 import hswish
|
||||
from ngraph.opset8 import idft
|
||||
from ngraph.opset8 import interpolate
|
||||
from ngraph.opset8 import less
|
||||
from ngraph.opset8 import less_equal
|
||||
from ngraph.opset8 import log
|
||||
from ngraph.opset8 import logical_and
|
||||
from ngraph.opset8 import logical_not
|
||||
from ngraph.opset8 import logical_or
|
||||
from ngraph.opset8 import logical_xor
|
||||
from ngraph.opset8 import log_softmax
|
||||
from ngraph.opset8 import loop
|
||||
from ngraph.opset8 import lrn
|
||||
from ngraph.opset8 import lstm_cell
|
||||
from ngraph.opset8 import lstm_sequence
|
||||
from ngraph.opset8 import matmul
|
||||
from ngraph.opset8 import max_pool
|
||||
from ngraph.opset8 import maximum
|
||||
from ngraph.opset8 import minimum
|
||||
from ngraph.opset8 import mish
|
||||
from ngraph.opset8 import mod
|
||||
from ngraph.opset8 import multiply
|
||||
from ngraph.opset8 import mvn
|
||||
from ngraph.opset8 import negative
|
||||
from ngraph.opset8 import non_max_suppression
|
||||
from ngraph.opset8 import non_zero
|
||||
from ngraph.opset8 import normalize_l2
|
||||
from ngraph.opset8 import not_equal
|
||||
from ngraph.opset8 import one_hot
|
||||
from ngraph.opset8 import pad
|
||||
from ngraph.opset8 import parameter
|
||||
from ngraph.opset8 import power
|
||||
from ngraph.opset8 import prelu
|
||||
from ngraph.opset8 import prior_box
|
||||
from ngraph.opset8 import prior_box_clustered
|
||||
from ngraph.opset8 import psroi_pooling
|
||||
from ngraph.opset8 import proposal
|
||||
from ngraph.opset8 import range
|
||||
from ngraph.opset8 import read_value
|
||||
from ngraph.opset8 import reduce_l1
|
||||
from ngraph.opset8 import reduce_l2
|
||||
from ngraph.opset8 import reduce_logical_and
|
||||
from ngraph.opset8 import reduce_logical_or
|
||||
from ngraph.opset8 import reduce_max
|
||||
from ngraph.opset8 import reduce_mean
|
||||
from ngraph.opset8 import reduce_min
|
||||
from ngraph.opset8 import reduce_prod
|
||||
from ngraph.opset8 import reduce_sum
|
||||
from ngraph.opset8 import region_yolo
|
||||
from ngraph.opset8 import reorg_yolo
|
||||
from ngraph.opset8 import relu
|
||||
from ngraph.opset8 import reshape
|
||||
from ngraph.opset8 import result
|
||||
from ngraph.opset8 import reverse_sequence
|
||||
from ngraph.opset8 import rnn_cell
|
||||
from ngraph.opset8 import rnn_sequence
|
||||
from ngraph.opset8 import roi_align
|
||||
from ngraph.opset8 import roi_pooling
|
||||
from ngraph.opset8 import roll
|
||||
from ngraph.opset8 import round
|
||||
from ngraph.opset8 import scatter_elements_update
|
||||
from ngraph.opset8 import scatter_update
|
||||
from ngraph.opset8 import select
|
||||
from ngraph.opset8 import selu
|
||||
from ngraph.opset8 import shape_of
|
||||
from ngraph.opset8 import shuffle_channels
|
||||
from ngraph.opset8 import sigmoid
|
||||
from ngraph.opset8 import sign
|
||||
from ngraph.opset8 import sin
|
||||
from ngraph.opset8 import sinh
|
||||
from ngraph.opset8 import softmax
|
||||
from ngraph.opset8 import softplus
|
||||
from ngraph.opset8 import space_to_batch
|
||||
from ngraph.opset8 import space_to_depth
|
||||
from ngraph.opset8 import split
|
||||
from ngraph.opset8 import sqrt
|
||||
from ngraph.opset8 import squared_difference
|
||||
from ngraph.opset8 import squeeze
|
||||
from ngraph.opset8 import strided_slice
|
||||
from ngraph.opset8 import subtract
|
||||
from ngraph.opset8 import swish
|
||||
from ngraph.opset8 import tan
|
||||
from ngraph.opset8 import tanh
|
||||
from ngraph.opset8 import tensor_iterator
|
||||
from ngraph.opset8 import tile
|
||||
from ngraph.opset8 import topk
|
||||
from ngraph.opset8 import transpose
|
||||
from ngraph.opset8 import unsqueeze
|
||||
from ngraph.opset8 import variadic_split
|
||||
|
||||
|
||||
# Extend Node class to support binary operators
|
||||
|
156
ngraph/python/src/ngraph/opset8/__init__.py
Normal file
156
ngraph/python/src/ngraph/opset8/__init__.py
Normal file
@ -0,0 +1,156 @@
|
||||
# Copyright (C) 2018-2021 Intel Corporation
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
from ngraph.opset1.ops import absolute
|
||||
from ngraph.opset1.ops import absolute as abs
|
||||
from ngraph.opset1.ops import acos
|
||||
from ngraph.opset4.ops import acosh
|
||||
from ngraph.opset1.ops import add
|
||||
from ngraph.opset1.ops import asin
|
||||
from ngraph.opset4.ops import asinh
|
||||
from ngraph.opset3.ops import assign
|
||||
from ngraph.opset1.ops import atan
|
||||
from ngraph.opset4.ops import atanh
|
||||
from ngraph.opset1.ops import avg_pool
|
||||
from ngraph.opset5.ops import batch_norm_inference
|
||||
from ngraph.opset2.ops import batch_to_space
|
||||
from ngraph.opset1.ops import binary_convolution
|
||||
from ngraph.opset3.ops import broadcast
|
||||
from ngraph.opset3.ops import bucketize
|
||||
from ngraph.opset1.ops import ceiling
|
||||
from ngraph.opset1.ops import ceiling as ceil
|
||||
from ngraph.opset1.ops import clamp
|
||||
from ngraph.opset1.ops import concat
|
||||
from ngraph.opset1.ops import constant
|
||||
from ngraph.opset1.ops import convert
|
||||
from ngraph.opset1.ops import convert_like
|
||||
from ngraph.opset1.ops import convolution
|
||||
from ngraph.opset1.ops import convolution_backprop_data
|
||||
from ngraph.opset1.ops import cos
|
||||
from ngraph.opset1.ops import cosh
|
||||
from ngraph.opset1.ops import ctc_greedy_decoder
|
||||
from ngraph.opset6.ops import ctc_greedy_decoder_seq_len
|
||||
from ngraph.opset4.ops import ctc_loss
|
||||
from ngraph.opset3.ops import cum_sum
|
||||
from ngraph.opset3.ops import cum_sum as cumsum
|
||||
from ngraph.opset1.ops import deformable_convolution
|
||||
from ngraph.opset1.ops import deformable_psroi_pooling
|
||||
from ngraph.opset1.ops import depth_to_space
|
||||
from ngraph.opset1.ops import detection_output
|
||||
from ngraph.opset7.ops import dft
|
||||
from ngraph.opset1.ops import divide
|
||||
from ngraph.opset7.ops import einsum
|
||||
from ngraph.opset1.ops import elu
|
||||
from ngraph.opset3.ops import embedding_bag_offsets_sum
|
||||
from ngraph.opset3.ops import embedding_bag_packed_sum
|
||||
from ngraph.opset3.ops import embedding_segments_sum
|
||||
from ngraph.opset3.ops import extract_image_patches
|
||||
from ngraph.opset1.ops import equal
|
||||
from ngraph.opset1.ops import erf
|
||||
from ngraph.opset1.ops import exp
|
||||
from ngraph.opset1.ops import fake_quantize
|
||||
from ngraph.opset1.ops import floor
|
||||
from ngraph.opset1.ops import floor_mod
|
||||
from ngraph.opset7.ops import gather
|
||||
from ngraph.opset6.ops import gather_elements
|
||||
from ngraph.opset5.ops import gather_nd
|
||||
from ngraph.opset1.ops import gather_tree
|
||||
from ngraph.opset7.ops import gelu
|
||||
from ngraph.opset1.ops import greater
|
||||
from ngraph.opset1.ops import greater_equal
|
||||
from ngraph.opset1.ops import grn
|
||||
from ngraph.opset1.ops import group_convolution
|
||||
from ngraph.opset1.ops import group_convolution_backprop_data
|
||||
from ngraph.opset3.ops import gru_cell
|
||||
from ngraph.opset5.ops import gru_sequence
|
||||
from ngraph.opset1.ops import hard_sigmoid
|
||||
from ngraph.opset5.ops import hsigmoid
|
||||
from ngraph.opset4.ops import hswish
|
||||
from ngraph.opset7.ops import idft
|
||||
from ngraph.opset1.ops import interpolate
|
||||
from ngraph.opset1.ops import less
|
||||
from ngraph.opset1.ops import less_equal
|
||||
from ngraph.opset1.ops import log
|
||||
from ngraph.opset1.ops import logical_and
|
||||
from ngraph.opset1.ops import logical_not
|
||||
from ngraph.opset1.ops import logical_or
|
||||
from ngraph.opset1.ops import logical_xor
|
||||
from ngraph.opset5.ops import log_softmax
|
||||
from ngraph.opset5.ops import loop
|
||||
from ngraph.opset1.ops import lrn
|
||||
from ngraph.opset4.ops import lstm_cell
|
||||
from ngraph.opset1.ops import lstm_sequence
|
||||
from ngraph.opset1.ops import matmul
|
||||
from ngraph.opset1.ops import max_pool
|
||||
from ngraph.opset1.ops import maximum
|
||||
from ngraph.opset1.ops import minimum
|
||||
from ngraph.opset4.ops import mish
|
||||
from ngraph.opset1.ops import mod
|
||||
from ngraph.opset1.ops import multiply
|
||||
from ngraph.opset6.ops import mvn
|
||||
from ngraph.opset1.ops import negative
|
||||
from ngraph.opset5.ops import non_max_suppression
|
||||
from ngraph.opset3.ops import non_zero
|
||||
from ngraph.opset1.ops import normalize_l2
|
||||
from ngraph.opset1.ops import not_equal
|
||||
from ngraph.opset1.ops import one_hot
|
||||
from ngraph.opset1.ops import pad
|
||||
from ngraph.opset1.ops import parameter
|
||||
from ngraph.opset1.ops import power
|
||||
from ngraph.opset1.ops import prelu
|
||||
from ngraph.opset1.ops import prior_box
|
||||
from ngraph.opset1.ops import prior_box_clustered
|
||||
from ngraph.opset1.ops import psroi_pooling
|
||||
from ngraph.opset4.ops import proposal
|
||||
from ngraph.opset1.ops import range
|
||||
from ngraph.opset3.ops import read_value
|
||||
from ngraph.opset4.ops import reduce_l1
|
||||
from ngraph.opset4.ops import reduce_l2
|
||||
from ngraph.opset1.ops import reduce_logical_and
|
||||
from ngraph.opset1.ops import reduce_logical_or
|
||||
from ngraph.opset1.ops import reduce_max
|
||||
from ngraph.opset1.ops import reduce_mean
|
||||
from ngraph.opset1.ops import reduce_min
|
||||
from ngraph.opset1.ops import reduce_prod
|
||||
from ngraph.opset1.ops import reduce_sum
|
||||
from ngraph.opset1.ops import region_yolo
|
||||
from ngraph.opset2.ops import reorg_yolo
|
||||
from ngraph.opset1.ops import relu
|
||||
from ngraph.opset1.ops import reshape
|
||||
from ngraph.opset1.ops import result
|
||||
from ngraph.opset1.ops import reverse_sequence
|
||||
from ngraph.opset3.ops import rnn_cell
|
||||
from ngraph.opset5.ops import rnn_sequence
|
||||
from ngraph.opset3.ops import roi_align
|
||||
from ngraph.opset2.ops import roi_pooling
|
||||
from ngraph.opset7.ops import roll
|
||||
from ngraph.opset5.ops import round
|
||||
from ngraph.opset3.ops import scatter_elements_update
|
||||
from ngraph.opset3.ops import scatter_update
|
||||
from ngraph.opset1.ops import select
|
||||
from ngraph.opset1.ops import selu
|
||||
from ngraph.opset3.ops import shape_of
|
||||
from ngraph.opset3.ops import shuffle_channels
|
||||
from ngraph.opset1.ops import sigmoid
|
||||
from ngraph.opset1.ops import sign
|
||||
from ngraph.opset1.ops import sin
|
||||
from ngraph.opset1.ops import sinh
|
||||
from ngraph.opset1.ops import softmax
|
||||
from ngraph.opset4.ops import softplus
|
||||
from ngraph.opset2.ops import space_to_batch
|
||||
from ngraph.opset1.ops import space_to_depth
|
||||
from ngraph.opset1.ops import split
|
||||
from ngraph.opset1.ops import sqrt
|
||||
from ngraph.opset1.ops import squared_difference
|
||||
from ngraph.opset1.ops import squeeze
|
||||
from ngraph.opset1.ops import strided_slice
|
||||
from ngraph.opset1.ops import subtract
|
||||
from ngraph.opset4.ops import swish
|
||||
from ngraph.opset1.ops import tan
|
||||
from ngraph.opset1.ops import tanh
|
||||
from ngraph.opset1.ops import tensor_iterator
|
||||
from ngraph.opset1.ops import tile
|
||||
from ngraph.opset3.ops import topk
|
||||
from ngraph.opset1.ops import transpose
|
||||
from ngraph.opset1.ops import unsqueeze
|
||||
from ngraph.opset1.ops import variadic_split
|
45
ngraph/python/src/ngraph/opset8/ops.py
Normal file
45
ngraph/python/src/ngraph/opset8/ops.py
Normal file
@ -0,0 +1,45 @@
|
||||
# Copyright (C) 2018-2021 Intel Corporation
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
"""Factory functions for all ngraph ops."""
|
||||
from functools import partial
|
||||
from typing import Callable, Iterable, List, Optional, Set, Union
|
||||
|
||||
import numpy as np
|
||||
from ngraph.impl import Node, Shape
|
||||
from ngraph.impl.op import Constant, Parameter
|
||||
from ngraph.opset_utils import _get_node_factory
|
||||
from ngraph.utils.decorators import binary_op, nameable_op, unary_op
|
||||
from ngraph.utils.input_validation import (
|
||||
assert_list_of_ints,
|
||||
check_valid_attributes,
|
||||
is_non_negative_value,
|
||||
is_positive_value,
|
||||
)
|
||||
from ngraph.utils.node_factory import NodeFactory
|
||||
from ngraph.utils.tensor_iterator_types import (
|
||||
GraphBody,
|
||||
TensorIteratorSliceInputDesc,
|
||||
TensorIteratorMergedInputDesc,
|
||||
TensorIteratorInvariantInputDesc,
|
||||
TensorIteratorBodyOutputDesc,
|
||||
TensorIteratorConcatOutputDesc,
|
||||
)
|
||||
from ngraph.utils.types import (
|
||||
NodeInput,
|
||||
NumericData,
|
||||
NumericType,
|
||||
ScalarData,
|
||||
TensorShape,
|
||||
as_node,
|
||||
as_nodes,
|
||||
get_dtype,
|
||||
get_element_type,
|
||||
get_element_type_str,
|
||||
make_constant_node,
|
||||
)
|
||||
|
||||
_get_node_factory_opset8 = partial(_get_node_factory, "opset8")
|
||||
|
||||
|
||||
# -------------------------------------------- ops ------------------------------------------------
|
@ -8,7 +8,7 @@ from _pyngraph import NodeFactory as _NodeFactory
|
||||
|
||||
from ngraph.impl import Node, Output
|
||||
|
||||
DEFAULT_OPSET = "opset6"
|
||||
DEFAULT_OPSET = "opset8"
|
||||
|
||||
|
||||
class NodeFactory(object):
|
||||
|
@ -75,6 +75,7 @@ namespace
|
||||
{"opset5", OpsetFunction(ngraph::get_opset5)},
|
||||
{"opset6", OpsetFunction(ngraph::get_opset6)},
|
||||
{"opset7", OpsetFunction(ngraph::get_opset7)},
|
||||
{"opset8", OpsetFunction(ngraph::get_opset8)},
|
||||
};
|
||||
|
||||
auto it = s_opsets.find(opset_ver);
|
||||
|
Loading…
Reference in New Issue
Block a user