1# Copyright 2020 Huawei Technologies Co., Ltd 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================ 15"""vm interface.""" 16# pylint: disable=wildcard-import,unused-wildcard-import 17from .vm_me import * 18 19 20class Vm: 21 pass 22 23 24vm = Vm() 25setattr(vm, 'avg_pooling', avg_pooling) 26setattr(vm, 'avg_pool_grad', avg_pool_grad) 27setattr(vm, 'batch_norm', batch_norm) 28setattr(vm, 'batch_norm_grad', batch_norm_grad) 29setattr(vm, 'col2im', col2im) 30setattr(vm, 'convolve', convolve) 31setattr(vm, 'conv2d', conv2d) 32setattr(vm, 'conv2d_backprop_filter', conv2d_backprop_filter) 33setattr(vm, 'conv2d_backprop_input', conv2d_backprop_input) 34setattr(vm, 'flatten', flatten) 35setattr(vm, 'flatten2', flatten2) 36setattr(vm, 'flatten_batch', flatten_batch) 37setattr(vm, 'flatten_grad', flatten_grad) 38setattr(vm, 'im2col', im2col) 39setattr(vm, 'matmul', matmul) 40setattr(vm, 'max_pooling', max_pooling) 41setattr(vm, 'max_pool_grad', max_pool_grad) 42setattr(vm, 'max_pool_grad_with_argmax', max_pool_grad_with_argmax) 43setattr(vm, 'max_pool_with_argmax', max_pool_with_argmax) 44setattr(vm, 'relu', relu) 45setattr(vm, 'relu_grad', relu_grad) 46setattr(vm, 'softmax', softmax) 47setattr(vm, 'softmax_cross_entropy_with_logits', softmax_cross_entropy_with_logits) 48setattr(vm, 'expand_dims', expand_dims) 49setattr(vm, 'squeeze', squeeze) 50setattr(vm, 'reshape', reshape) 51setattr(vm, 'shape', shape) 52setattr(vm, 'rank', rank) 53setattr(vm, 'logsoftmax', logsoftmax) 54setattr(vm, 'transpose', transpose) 55setattr(vm, 'invert_permutation', invert_permutation) 56setattr(vm, 'select', select) 57setattr(vm, 'sum', sum_by_axis) 58setattr(vm, 'equal', equal) 59setattr(vm, 'not_equal', not_equal) 60setattr(vm, 'greater', greater) 61setattr(vm, 'less', less) 62setattr(vm, 'logical_not', logical_not) 63setattr(vm, 'sqrt', sqrt) 64setattr(vm, 'power', power) 65setattr(vm, "exp", exp) 66setattr(vm, "tanh", tanh) 67setattr(vm, "sigmoid", sigmoid) 68setattr(vm, 'maximum', maximum) 69setattr(vm, 'minimum', minimum) 70setattr(vm, 'all', all_) 71setattr(vm, 'any', any_) 72