tvm_ffi/
lib.rs

1/*
2 * Licensed to the Apache Software Foundation (ASF) under one
3 * or more contributor license agreements.  See the NOTICE file
4 * distributed with this work for additional information
5 * regarding copyright ownership.  The ASF licenses this file
6 * to you under the Apache License, Version 2.0 (the
7 * "License"); you may not use this file except in compliance
8 * with the License.  You may obtain a copy of the License at
9 *
10 *   http://www.apache.org/licenses/LICENSE-2.0
11 *
12 * Unless required by applicable law or agreed to in writing,
13 * software distributed under the License is distributed on an
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15 * KIND, either express or implied.  See the License for the
16 * specific language governing permissions and limitations
17 * under the License.
18 */
19pub mod any;
20pub mod collections;
21pub mod derive;
22pub mod device;
23pub mod dtype;
24pub mod error;
25pub mod extra;
26pub mod function;
27pub mod function_internal;
28pub mod macros;
29#[doc(hidden)]
30pub mod match_any_internal;
31pub mod object;
32pub mod optional;
33pub mod reflection;
34pub mod rvalue_ref;
35pub mod string;
36pub mod type_traits;
37pub use tvm_ffi_sys;
38
39pub use crate::any::{Any, AnyView};
40pub use crate::collections::array::Array;
41pub use crate::collections::map::Map;
42pub use crate::collections::shape::Shape;
43pub use crate::collections::tensor::{CPUNDAlloc, NDAllocator, Tensor};
44pub use crate::device::{current_stream, with_stream};
45pub use crate::dtype::DLDataTypeExt;
46pub use crate::error::{Error, ErrorKind, Result};
47pub use crate::error::{
48    ATTRIBUTE_ERROR, INDEX_ERROR, INTERNAL_ERROR, KEY_ERROR, RUNTIME_ERROR, TYPE_ERROR, VALUE_ERROR,
49};
50pub use crate::extra::module::Module;
51pub use crate::extra::structural_mutate::{
52    structural_map, structural_mutate, DefaultMutContextPolicy, InplaceMode, InplaceValue,
53    IntoMapper, IntoMutator, MapChainLink, MapDispatch, MapWithContextPolicy, MutContextPolicy,
54    MutateCallbacks, MutateChainLink, MutateContext, MutateDispatch, MutateValue, Mutator,
55    StructuralMutator, StructuralVarRemap,
56};
57pub use crate::extra::structural_visit::{
58    structural_visit, structural_walk, ContextPolicy, DefRegionKind, DefaultContextPolicy,
59    IntoVisitor, IntoWalkResult, IntoWalker, StructuralVisitor, VisitCallbacks, VisitChainLink,
60    VisitContext, VisitInterrupt, WalkChainLink, WalkDispatch, WalkOrder, WalkResult,
61    WalkWithContextPolicy,
62};
63pub use crate::extra::unchanged::{Unchanged, UnchangedOr};
64pub use crate::extra::StructuralView;
65pub use crate::function::Function;
66pub use crate::object::ObjectRefCast;
67pub use crate::object::{
68    Object, ObjectArc, ObjectCore, ObjectCoreWithExtraItems, ObjectIdentity, ObjectRefCore,
69};
70pub use crate::optional::{Optional, OptionalCompatible};
71pub use crate::reflection::{get_type_attr, FieldGetter, TypeAttrColumn};
72pub use crate::rvalue_ref::RValueRef;
73pub use crate::string::{Bytes, String};
74pub use crate::type_traits::AnyCompatible;
75pub use tvm_ffi_macros::{dispatch, match_any};
76
77pub use tvm_ffi_sys::TVMFFITypeIndex as TypeIndex;
78pub use tvm_ffi_sys::{
79    DLDataType, DLDataTypeCode, DLDevice, DLDeviceType, TVMFFIAny, TVMFFIObject, TVMFFIStreamHandle,
80};
tvm_ffi/lib.rs

tvm_ffi/
lib.rs
