Index A | C | D | F | L | M | N | P | R | S | T | U A adaptiveavgpool2d_custom (C++ function) ai3 module AlgorithmicSelector (in module ai3) avgpool2d_custom (C++ function) C Context (C++ class) Context::cublas_handle_t (C++ function) Context::cudnn_handle_t (C++ function) Context::mps_graph (C++ function) Context::mps_graph_device (C++ function) Context::mtl_device (C++ function) Context::sycl_queue (C++ function) CONV2D_ALGOS_TO_USE (in module run) conv2d_custom (C++ function) D DEFAULT_ADAPTIVEAVGPOOL2D (C++ member) DEFAULT_AVGPOOL2D (C++ member) DEFAULT_CONV2D (C++ member) DEFAULT_FLATTEN (C++ member) DEFAULT_LINEAR (C++ member) DEFAULT_MAXPOOL2D (C++ member) DEFAULT_RELU (C++ member) F flatten_custom (C++ function) FROM_BACKEND (in module ai3) L linear_custom (C++ function) M maxpool2d_custom (C++ function) Model (class in ai3) module ai3 run N numpy() (ai3.Tensor method) P predict() (ai3.Model method) R relu_custom (C++ function) run module S sample_dims (C++ type) sample_dims::ACTIVATION (C++ member) sample_dims::CONV2D (C++ member) sample_dims::FLATTEN (C++ member) sample_dims::LINEAR (C++ member) sample_dims::POOL2D (C++ member) SUPPORTED_ALGORITHMS (in module ai3) SUPPORTED_FROM_BACKENDS (in module ai3) swap_backend() (in module ai3) swap_conv2d() (in module ai3) T Tensor (C++ class) (class in ai3) Tensor::batch_size (C++ function) Tensor::batched (C++ function) Tensor::buffer (C++ function) Tensor::count (C++ function) Tensor::data (C++ member) Tensor::form_tensor (C++ function) Tensor::from_optional (C++ function) Tensor::height (C++ function) Tensor::input_channels (C++ function) Tensor::output_channels (C++ function) Tensor::owned (C++ member) Tensor::scalar_type (C++ member) Tensor::shape (C++ member) Tensor::Tensor (C++ function), [1], [2] Tensor::to_type (C++ function) Tensor::width (C++ function) to() (ai3.Tensor method) torch() (ai3.Tensor method) U USE_ALL_POSSIBLE_CONV (in module run) using_cublas() (in module ai3) using_cudnn() (in module ai3) using_mps_and_metal() (in module ai3) using_sycl() (in module ai3)