max_utils
Structs
Functions
tensor_to_array
tensor_to_array(owned src: Tensor[float32]) -> Array
top_order
top_order(mut curr: Array) -> List[Array]
to_tensor
to_tensor(arg: Array) -> Tensor[float32]
make_equal_rank
make_equal_rank(arg: Symbol, arg_shape: List[Int], comp_shape: List[Int]) -> Symbol
build_graph
build_graph(args: List[Array], outputs: List[Array], trace: List[Array]) -> Graph
build_model
build_model(args: List[Array], outputs: List[Array], trace: List[Array]) -> Model
execute_model
execute_model(args: List[Array], outputs: List[Array], model: Model) -> List[Array]
Last updated on