1# Licensed to the Apache Software Foundation (ASF) under one 2# or more contributor license agreements. See the NOTICE file 3# distributed with this work for additional information 4# regarding copyright ownership. The ASF licenses this file 5# to you under the Apache License, Version 2.0 (the 6# "License"); you may not use this file except in compliance 7# with the License. You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, 12# software distributed under the License is distributed on an 13# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 14# KIND, either express or implied. See the License for the 15# specific language governing permissions and limitations 16# under the License. 17 18# pylint: disable=redefined-builtin, wildcard-import 19"""CUDA specific declaration and schedules.""" 20from __future__ import absolute_import as _abs 21 22from . import conv2d, depthwise_conv2d, conv2d_transpose_nchw, deformable_conv2d, \ 23 group_conv2d_nchw, dense 24from .conv2d_hwcn import schedule_conv2d_hwcn 25from .depthwise_conv2d import schedule_depthwise_conv2d_backward_input_nhwc 26from .depthwise_conv2d import schedule_depthwise_conv2d_backward_weight_nhwc 27from .group_conv2d_nchw import schedule_conv2d_nchw_cuda 28from .reduction import schedule_reduce 29from .softmax import schedule_softmax 30from .injective import schedule_injective, schedule_elemwise, schedule_broadcast 31from .dense import schedule_dense 32from .pooling import schedule_pool, schedule_adaptive_pool 33from .nn import schedule_lrn, schedule_l2_normalize 34from .batch_matmul import schedule_batch_matmul 35from .vision import * 36from . import ssd 37from .ssd import * 38from .nms import * 39from .rcnn import * 40from .sort import * 41