List of all items
Structs
- VersionInfo
- data::CsvDataset
- data::CsvDatasetBuilder
- data::DataLoader
- data::DataLoaderBuilder
- data::ImageFolderDataset
- data::ImageFolderDatasetBuilder
- data::RandomSampler
- nn::BatchNorm
- nn::Conv2D
- nn::Dense
- nn::Dropout
- nn::GRU
- nn::LSTM
- nn::MaxPool2D
- nn::MultiHeadAttention
- nn::RMSNorm
- nn::RNN
- nn::Sequential
- nn::TransformerDecoder
- nn::TransformerEncoder
- onnx::OnnxGraph
- onnx::OnnxModel
- onnx::OnnxNode
- onnx::OnnxTensor
- onnx::OnnxValueInfo
- optim::Adadelta
- optim::Adagrad
- optim::Adam
- optim::AdamW
- optim::LAMB
- optim::Lion
- optim::Lookahead
- optim::Nadam
- optim::ParameterGroup
- optim::ParameterGroupOptimizer
- optim::RAdam
- optim::RMSprop
- optim::SGD
- platform::SimdCapabilities
- prelude::Adam
- prelude::AdamW
- prelude::BatchNorm
- prelude::Conv2D
- prelude::CsvDataset
- prelude::CsvDatasetBuilder
- prelude::DataLoader
- prelude::DataLoaderBuilder
- prelude::Dense
- prelude::Dropout
- prelude::EarlyStopping
- prelude::GRU
- prelude::GradientTape
- prelude::ImageFolderDataset
- prelude::ImageFolderDatasetBuilder
- prelude::LSTM
- prelude::MaxPool2D
- prelude::ModelCheckpoint
- prelude::MultiHeadAttention
- prelude::ParameterGroup
- prelude::RMSNorm
- prelude::RNN
- prelude::RandomSampler
- prelude::SGD
- prelude::Sequential
- prelude::Tensor
- prelude::TrackedTensor
- prelude::Trainer
- prelude::TransformerDecoder
- prelude::TransformerEncoder
- prelude::ops::BroadcastableConstraint
- prelude::ops::ExactShapeConstraint
- prelude::ops::FrameworkBenchmarkConfig
- prelude::ops::FrameworkComparisonResult
- prelude::ops::FusionGraph
- prelude::ops::FusionNode
- prelude::ops::FusionPassBuilder
- prelude::ops::FusionStats
- prelude::ops::GpuBenchmarkConfig
- prelude::ops::GpuBenchmarkResult
- prelude::ops::GpuTensorFlowBenchmark
- prelude::ops::MatMulCompatibleConstraint
- prelude::ops::MinRankConstraint
- prelude::ops::RankConstraint
- prelude::ops::ShapeContext
- prelude::ops::ShapeValidator
- prelude::ops::UfuncRegistry
- prelude::ops::UltraPerformanceConfig
- prelude::ops::activation::ActivationAnalytics
- prelude::ops::activation::ActivationConfig
- prelude::ops::activation::ActivationRegistry
- prelude::ops::activation::GpuActivationStream
- prelude::ops::activation::core::ActivationAnalytics
- prelude::ops::activation::core::ActivationRegistry
- prelude::ops::activation::gpu::GpuActivationStream
- prelude::ops::activation::strategy::ActivationConfig
- prelude::ops::async_binary::AsyncBinaryOpFuture
- prelude::ops::async_binary::AsyncBinaryOperationExecutor
- prelude::ops::benchmark::BenchmarkConfig
- prelude::ops::benchmark::BenchmarkResult
- prelude::ops::benchmark::BenchmarkStatistics
- prelude::ops::benchmark::BenchmarkSuite
- prelude::ops::benchmark::RegressionReport
- prelude::ops::benchmark::Statistics
- prelude::ops::binary::AddOp
- prelude::ops::binary::BinaryOpAnalytics
- prelude::ops::binary::DivOp
- prelude::ops::binary::FusedAddMulOp
- prelude::ops::binary::FusedMulAddOp
- prelude::ops::binary::MaxOp
- prelude::ops::binary::MinOp
- prelude::ops::binary::MulOp
- prelude::ops::binary::PReLUOp
- prelude::ops::binary::PowOp
- prelude::ops::binary::SubOp
- prelude::ops::binary::core::BinaryOpAnalytics
- prelude::ops::binary::core::BinaryOpRegistry
- prelude::ops::binary::fused::fused_ops::FusedAddMulOp
- prelude::ops::binary::fused::fused_ops::FusedMulAddOp
- prelude::ops::binary::operations::AddOp
- prelude::ops::binary::operations::DivOp
- prelude::ops::binary::operations::MaxOp
- prelude::ops::binary::operations::MinOp
- prelude::ops::binary::operations::MulOp
- prelude::ops::binary::operations::PReLUOp
- prelude::ops::binary::operations::PowOp
- prelude::ops::binary::operations::SubOp
- prelude::ops::comparison::EqOp
- prelude::ops::comparison::GeOp
- prelude::ops::comparison::GtOp
- prelude::ops::comparison::LeOp
- prelude::ops::comparison::LtOp
- prelude::ops::comparison::NeOp
- prelude::ops::fft::gpu_kernels::FFTInfo
- prelude::ops::framework_comparison::FrameworkBenchmarkConfig
- prelude::ops::framework_comparison::FrameworkComparisonResult
- prelude::ops::fusion::FusionGraph
- prelude::ops::fusion::FusionNode
- prelude::ops::fusion::FusionPassBuilder
- prelude::ops::fusion::FusionStats
- prelude::ops::gpu_tensorflow_benchmark::GpuBenchmarkConfig
- prelude::ops::gpu_tensorflow_benchmark::GpuBenchmarkResult
- prelude::ops::gpu_tensorflow_benchmark::GpuTensorFlowBenchmark
- prelude::ops::hybrid_scheduler::HybridWorkFuture
- prelude::ops::hybrid_scheduler::HybridWorkScheduler
- prelude::ops::hybrid_scheduler::ResourceMetrics
- prelude::ops::hybrid_scheduler::SchedulingConfig
- prelude::ops::hybrid_scheduler::WorkItem
- prelude::ops::logical::AndOp
- prelude::ops::logical::NotOp
- prelude::ops::logical::OrOp
- prelude::ops::logical::XorOp
- prelude::ops::numpy_compat::BinaryUfunc
- prelude::ops::numpy_compat::UfuncRegistry
- prelude::ops::numpy_compat::UnaryUfunc
- prelude::ops::optimized_binary::OptimizedAddOp
- prelude::ops::optimized_binary::OptimizedDivOp
- prelude::ops::optimized_binary::OptimizedMulOp
- prelude::ops::optimized_binary::OptimizedSubOp
- prelude::ops::performance_benchmark::BenchmarkConfig
- prelude::ops::performance_benchmark::BenchmarkResult
- prelude::ops::reduction::gpu_kernels::GpuReductionConfig
- prelude::ops::reduction::gpu_kernels::MultiStageReduction
- prelude::ops::reduction::gpu_kernels::TreeReduction
- prelude::ops::reduction::gpu_kernels::WarpReduction
- prelude::ops::reduction::segment::SegmentConfig
- prelude::ops::reduction::segment::types::SegmentConfig
- prelude::ops::registry::ArgDef
- prelude::ops::registry::AttrDef
- prelude::ops::registry::BatchOperation
- prelude::ops::registry::HftRegistry
- prelude::ops::registry::OP_REGISTRY
- prelude::ops::registry::OpDef
- prelude::ops::registry::OpRegistry
- prelude::ops::registry::OpVersion
- prelude::ops::registry::PrecisionMetrics
- prelude::ops::registry::QuantumRegistry
- prelude::ops::registry::RegistryAnalytics
- prelude::ops::registry::core::OP_REGISTRY
- prelude::ops::registry::core::RegistryAnalytics
- prelude::ops::registry::types::ArgDef
- prelude::ops::registry::types::AttrDef
- prelude::ops::registry::types::BatchOperation
- prelude::ops::registry::types::OpDef
- prelude::ops::registry::types::OpRegistry
- prelude::ops::registry::types::OpVersion
- prelude::ops::registry::ultra_extensions::HftRegistry
- prelude::ops::registry::ultra_extensions::PrecisionMetrics
- prelude::ops::registry::ultra_extensions::QuantumRegistry
- prelude::ops::registry_extensions::DeviceCapabilities
- prelude::ops::registry_extensions::ENHANCED_REGISTRY
- prelude::ops::registry_extensions::EnhancedRegistry
- prelude::ops::registry_extensions::KernelStats
- prelude::ops::registry_extensions::PerformanceReport
- prelude::ops::shape_inference::BroadcastableConstraint
- prelude::ops::shape_inference::ExactShapeConstraint
- prelude::ops::shape_inference::MatMulCompatibleConstraint
- prelude::ops::shape_inference::MinRankConstraint
- prelude::ops::shape_inference::RankConstraint
- prelude::ops::shape_inference::ShapeContext
- prelude::ops::shape_inference::ShapeValidator
- prelude::ops::shape_inference_registry::ShapeInferenceRegistry
- prelude::ops::shape_inference_registry::registry::ShapeInferenceRegistry
- prelude::ops::stats::StatisticalMetrics
- prelude::ops::stats::StatsConfig
- prelude::ops::stats::config::StatisticalMetrics
- prelude::ops::stats::config::StatsConfig
- prelude::ops::ultra_matmul_v3::UltraPerformanceConfig
- prelude::ops::unified_dispatch::DispatchContext
- prelude::ops::unified_dispatch::DispatchStats
- prelude::ops::unified_dispatch::UnifiedDispatcher
- version_check::SubcrateVersion
- version_check::VersionConsistencyReport
Enums
- TensorError
- error::FrameworkError
- logging::LogLevel
- nn::ActivationFunction
- onnx::OnnxAttribute
- onnx::OnnxDataType
- onnx::OnnxError
- onnx::OnnxFormat
- platform::Platform
- prelude::ActivationFunction
- prelude::DType
- prelude::Device
- prelude::TensorError
- prelude::dtype::DType
- prelude::ops::ElementwiseOpType
- prelude::ops::activation::ActivationStrategy
- prelude::ops::activation::GpuActivationOp
- prelude::ops::activation::core::ActivationStrategy
- prelude::ops::activation::gpu::GpuActivationOp
- prelude::ops::activation::strategy::ActivationStrategy
- prelude::ops::async_binary::WorkPriority
- prelude::ops::benchmark::RegressionSeverity
- prelude::ops::binary::OpComplexity
- prelude::ops::binary::core::OpComplexity
- prelude::ops::fusion::ElementwiseOpType
- prelude::ops::hybrid_scheduler::ExecutionStrategy
- prelude::ops::hybrid_scheduler::WorkPriority
- prelude::ops::hybrid_scheduler::WorkType
- prelude::ops::matmul::specialized::MixedPrecisionMode
- prelude::ops::reduction::gpu_execution::GpuReductionOp
- prelude::ops::reduction::gpu_kernels::ReductionOp
- prelude::ops::registry::AttrType
- prelude::ops::registry::AttrValue
- prelude::ops::registry::types::AttrType
- prelude::ops::registry::types::AttrValue
- prelude::ops::registry_extensions::KernelSelectionStrategy
- prelude::ops::shape_inference_registry::MetadataValue
- prelude::ops::shape_inference_registry::OperationCategory
- prelude::ops::shape_inference_registry::types::MetadataValue
- prelude::ops::shape_inference_registry::types::OperationCategory
- prelude::ops::unified_dispatch::Backend
Traits
- data::Dataset
- nn::Layer
- nn::Model
- onnx::OnnxExport
- onnx::OnnxImport
- optim::Optimizer
- prelude::Dataset
- prelude::Layer
- prelude::Model
- prelude::Optimizer
- prelude::ops::ShapeConstraint
- prelude::ops::activation::ActivationFunction
- prelude::ops::activation::core::ActivationFunction
- prelude::ops::binary::BinaryOp
- prelude::ops::binary::core::BinaryOp
- prelude::ops::comparison::ComparisonOp
- prelude::ops::logical::LogicalOp
- prelude::ops::logical::UnaryLogicalOp
- prelude::ops::numpy_compat::UniversalFunction
- prelude::ops::optimized_binary::OptimizedBinaryOp
- prelude::ops::registry::Kernel
- prelude::ops::registry::UltraKernel
- prelude::ops::registry::core::UltraKernel
- prelude::ops::registry::types::Kernel
- prelude::ops::shape_inference::ShapeConstraint
Macros
Functions
- interop::ndarray::from_ndarray
- interop::ndarray::from_slice_with_shape
- interop::ndarray::round_trip
- interop::ndarray::to_ndarray
- io::load_tensor
- io::save_tensor
- logging::current_log_level
- logging::emit
- logging::init_from_env
- logging::set_log_backend
- logging::set_log_level
- platform::current_platform
- platform::detect_simd_capabilities
- prelude::binary_cross_entropy
- prelude::categorical_cross_entropy
- prelude::dtype::dtype_from_type
- prelude::mse
- prelude::ops::abs
- prelude::ops::absolute
- prelude::ops::activation::activation_module_info
- prelude::ops::activation::batch_gpu_activations
- prelude::ops::activation::core::get_activation_performance_report
- prelude::ops::activation::core::get_activation_registry
- prelude::ops::activation::core::reset_activation_counters
- prelude::ops::activation::dispatch_gpu_activation
- prelude::ops::activation::elu
- prelude::ops::activation::execute_gpu_activation
- prelude::ops::activation::fast_sigmoid_approx
- prelude::ops::activation::functions::elu
- prelude::ops::activation::functions::gelu
- prelude::ops::activation::functions::gelu_f32
- prelude::ops::activation::functions::hard_swish
- prelude::ops::activation::functions::hardswish
- prelude::ops::activation::functions::leaky_relu
- prelude::ops::activation::functions::log_softmax
- prelude::ops::activation::functions::mish
- prelude::ops::activation::functions::prelu
- prelude::ops::activation::functions::relu
- prelude::ops::activation::functions::relu6
- prelude::ops::activation::functions::relu_f32
- prelude::ops::activation::functions::relu_f64
- prelude::ops::activation::functions::sigmoid
- prelude::ops::activation::functions::sigmoid_f32
- prelude::ops::activation::functions::sigmoid_f64
- prelude::ops::activation::functions::softmax
- prelude::ops::activation::functions::swish
- prelude::ops::activation::functions::tanh
- prelude::ops::activation::functions::tanh_f32
- prelude::ops::activation::functions::tanh_f64
- prelude::ops::activation::gelu
- prelude::ops::activation::gelu_f32
- prelude::ops::activation::gelu_parallel_f32
- prelude::ops::activation::gelu_sequential_f32
- prelude::ops::activation::get_activation_performance_report
- prelude::ops::activation::get_activation_registry
- prelude::ops::activation::gpu::batch_gpu_activations
- prelude::ops::activation::gpu::dispatch_gpu_activation
- prelude::ops::activation::gpu::execute_gpu_activation
- prelude::ops::activation::gpu::softmax_gpu_f32
- prelude::ops::activation::gpu_available
- prelude::ops::activation::hard_swish
- prelude::ops::activation::hardswish
- prelude::ops::activation::implementations::gelu_parallel_f32
- prelude::ops::activation::implementations::gelu_sequential_f32
- prelude::ops::activation::implementations::sigmoid_vectorized
- prelude::ops::activation::implementations::tanh_vectorized
- prelude::ops::activation::implementations::ultra_relu_vectorized
- prelude::ops::activation::implementations::ultra_sigmoid_vectorized
- prelude::ops::activation::leaky_relu
- prelude::ops::activation::log_softmax
- prelude::ops::activation::mish
- prelude::ops::activation::prelu
- prelude::ops::activation::relu
- prelude::ops::activation::relu6
- prelude::ops::activation::relu_f32
- prelude::ops::activation::relu_f64
- prelude::ops::activation::reset_activation_counters
- prelude::ops::activation::select_activation_strategy
- prelude::ops::activation::sigmoid
- prelude::ops::activation::sigmoid_f32
- prelude::ops::activation::sigmoid_f64
- prelude::ops::activation::sigmoid_vectorized
- prelude::ops::activation::simd::fast_sigmoid_approx
- prelude::ops::activation::simd::simd_gelu_f32
- prelude::ops::activation::simd::simd_relu_f32
- prelude::ops::activation::simd::simd_sigmoid_f32
- prelude::ops::activation::simd::simd_tanh_f32
- prelude::ops::activation::simd_available
- prelude::ops::activation::simd_gelu_f32
- prelude::ops::activation::simd_relu_f32
- prelude::ops::activation::simd_sigmoid_f32
- prelude::ops::activation::simd_tanh_f32
- prelude::ops::activation::softmax
- prelude::ops::activation::softmax_gpu_f32
- prelude::ops::activation::strategy::select_activation_strategy
- prelude::ops::activation::supported_activation_count
- prelude::ops::activation::swish
- prelude::ops::activation::tanh
- prelude::ops::activation::tanh_f32
- prelude::ops::activation::tanh_f64
- prelude::ops::activation::tanh_vectorized
- prelude::ops::activation::ultra_relu_vectorized
- prelude::ops::activation::ultra_sigmoid_vectorized
- prelude::ops::activation::version
- prelude::ops::adaptive_avg_pool2d
- prelude::ops::adaptive_max_pool2d
- prelude::ops::add
- prelude::ops::add_async
- prelude::ops::add_async_priority
- prelude::ops::advanced_math::expit
- prelude::ops::advanced_math::gelu_tanh
- prelude::ops::advanced_math::hard_sigmoid
- prelude::ops::advanced_math::hard_swish
- prelude::ops::advanced_math::log_sigmoid
- prelude::ops::advanced_math::logit
- prelude::ops::advanced_math::logsumexp
- prelude::ops::advanced_math::mish
- prelude::ops::advanced_math::selu
- prelude::ops::advanced_math::softplus
- prelude::ops::advanced_math::softsign
- prelude::ops::apply_ufunc
- prelude::ops::arange
- prelude::ops::arccos
- prelude::ops::arccosh
- prelude::ops::arcsin
- prelude::ops::arcsinh
- prelude::ops::arctan
- prelude::ops::arctan2
- prelude::ops::arctanh
- prelude::ops::argmax
- prelude::ops::argmin
- prelude::ops::async_binary::add_async
- prelude::ops::async_binary::add_async_priority
- prelude::ops::async_binary::batch_add_async
- prelude::ops::async_binary::batch_mul_async
- prelude::ops::async_binary::div_async
- prelude::ops::async_binary::global_async_executor
- prelude::ops::async_binary::is_async_operations_idle
- prelude::ops::async_binary::mul_async
- prelude::ops::async_binary::mul_async_priority
- prelude::ops::async_binary::pow_async
- prelude::ops::async_binary::prelu_async
- prelude::ops::async_binary::sub_async
- prelude::ops::async_binary::synchronize_async_operations
- prelude::ops::avg_pool2d
- prelude::ops::avg_pool3d
- prelude::ops::basic::add
- prelude::ops::basic::clamp
- prelude::ops::basic::div
- prelude::ops::basic::mul
- prelude::ops::basic::scalar_add
- prelude::ops::basic::sub
- prelude::ops::batch_add_async
- prelude::ops::batch_matmul
- prelude::ops::batch_mul_async
- prelude::ops::batch_norm
- prelude::ops::benchmark::benchmark_binary_op
- prelude::ops::benchmark::benchmark_matmul_sizes
- prelude::ops::benchmark::benchmark_unary_op
- prelude::ops::bessel_j0
- prelude::ops::bessel_j1
- prelude::ops::bessel_y0
- prelude::ops::bessel_y1
- prelude::ops::binary::add
- prelude::ops::binary::binary_op
- prelude::ops::binary::broadcast_array
- prelude::ops::binary::clamp
- prelude::ops::binary::convenience::add
- prelude::ops::binary::convenience::clamp
- prelude::ops::binary::convenience::div
- prelude::ops::binary::convenience::get_binary_op_performance_report
- prelude::ops::binary::convenience::max
- prelude::ops::binary::convenience::min
- prelude::ops::binary::convenience::mul
- prelude::ops::binary::convenience::pow
- prelude::ops::binary::convenience::reset_binary_op_counters
- prelude::ops::binary::convenience::scalar_add
- prelude::ops::binary::convenience::sub
- prelude::ops::binary::convenience::ultra_add
- prelude::ops::binary::convenience::ultra_div
- prelude::ops::binary::convenience::ultra_mul
- prelude::ops::binary::convenience::ultra_sub
- prelude::ops::binary::core::get_binary_op_registry
- prelude::ops::binary::div
- prelude::ops::binary::fused::prefetch_memory
- prelude::ops::binary::get_binary_op_performance_report
- prelude::ops::binary::get_binary_op_registry
- prelude::ops::binary::implementation::binary_op
- prelude::ops::binary::max
- prelude::ops::binary::min
- prelude::ops::binary::mul
- prelude::ops::binary::pow
- prelude::ops::binary::reset_binary_op_counters
- prelude::ops::binary::scalar_add
- prelude::ops::binary::simd::simd_f32_ops::simd_add_f32
- prelude::ops::binary::simd::simd_f32_ops::simd_mul_f32
- prelude::ops::binary::sub
- prelude::ops::binary::ultra_add
- prelude::ops::binary::ultra_div
- prelude::ops::binary::ultra_mul
- prelude::ops::binary::ultra_sub
- prelude::ops::broadcast_to
- prelude::ops::cast
- prelude::ops::ceil
- prelude::ops::cholesky
- prelude::ops::cholesky_lapack
- prelude::ops::clamp
- prelude::ops::clear_performance_analytics
- prelude::ops::comparison::comparison_op
- prelude::ops::comparison::eq
- prelude::ops::comparison::ge
- prelude::ops::comparison::gt
- prelude::ops::comparison::le
- prelude::ops::comparison::lt
- prelude::ops::comparison::ne
- prelude::ops::concat
- prelude::ops::configure_ultra_performance
- prelude::ops::conv1d
- prelude::ops::conv2d
- prelude::ops::conv3d
- prelude::ops::conv::conv1d
- prelude::ops::conv::conv1d::conv1d
- prelude::ops::conv::conv2d
- prelude::ops::conv::conv2d::conv2d
- prelude::ops::conv::conv2d::conv2d_auto_layout
- prelude::ops::conv::conv2d::conv2d_with_layout
- prelude::ops::conv::conv2d_auto_layout
- prelude::ops::conv::conv2d_with_layout
- prelude::ops::conv::conv3d
- prelude::ops::conv::conv3d::conv3d
- prelude::ops::conv::conv_layout_benchmark
- prelude::ops::conv::conv_transpose2d
- prelude::ops::conv::depthwise::depthwise_conv2d
- prelude::ops::conv::depthwise_conv2d
- prelude::ops::conv::layout::conv2d_auto_layout
- prelude::ops::conv::layout::conv2d_with_layout
- prelude::ops::conv::layout::conv_layout_benchmark
- prelude::ops::conv::layout::layout_conversion_cost
- prelude::ops::conv::layout::select_optimal_layout
- prelude::ops::conv::layout::should_convert_layout
- prelude::ops::conv::transpose::conv_transpose2d
- prelude::ops::correlation
- prelude::ops::cos
- prelude::ops::cosh
- prelude::ops::covariance
- prelude::ops::create_f32_ufunc_registry
- prelude::ops::cumprod
- prelude::ops::cumsum
- prelude::ops::depthwise_conv2d
- prelude::ops::det
- prelude::ops::determinant_lapack
- prelude::ops::diag
- prelude::ops::diagonal
- prelude::ops::digamma
- prelude::ops::div
- prelude::ops::div_async
- prelude::ops::dot
- prelude::ops::eig
- prelude::ops::eigenvalues_lapack
- prelude::ops::einsum
- prelude::ops::einsum::batch_transpose
- prelude::ops::einsum::blas::try_blas_optimized_patterns
- prelude::ops::einsum::cache::cache_optimized_contraction
- prelude::ops::einsum::cache::execute_contraction_path
- prelude::ops::einsum::cache_friendly_trace
- prelude::ops::einsum::cache_optimized_contraction
- prelude::ops::einsum::compute_outer_product
- prelude::ops::einsum::core::einsum
- prelude::ops::einsum::core::extract_diagonal
- prelude::ops::einsum::core::infer_output_subscript
- prelude::ops::einsum::core::parse_einsum_equation
- prelude::ops::einsum::einsum
- prelude::ops::einsum::execute_contraction_path
- prelude::ops::einsum::extract_diagonal
- prelude::ops::einsum::gpu::gpu_einsum_batched_matmul
- prelude::ops::einsum::gpu::gpu_einsum_diagonal
- prelude::ops::einsum::gpu::gpu_einsum_matmul
- prelude::ops::einsum::gpu::gpu_einsum_outer_product
- prelude::ops::einsum::gpu::gpu_einsum_trace
- prelude::ops::einsum::gpu::gpu_einsum_transpose
- prelude::ops::einsum::gpu::gpu_einsum_vector_dot
- prelude::ops::einsum::gpu_einsum_batched_matmul
- prelude::ops::einsum::gpu_einsum_diagonal
- prelude::ops::einsum::gpu_einsum_matmul
- prelude::ops::einsum::gpu_einsum_outer_product
- prelude::ops::einsum::gpu_einsum_trace
- prelude::ops::einsum::gpu_einsum_transpose
- prelude::ops::einsum::gpu_einsum_vector_dot
- prelude::ops::einsum::infer_output_subscript
- prelude::ops::einsum::parse_einsum_equation
- prelude::ops::einsum::patterns::try_optimize_common_patterns
- prelude::ops::einsum::patterns::try_optimize_gpu_patterns
- prelude::ops::einsum::try_blas_optimized_patterns
- prelude::ops::einsum::try_optimize_common_patterns
- prelude::ops::einsum::try_optimize_gpu_patterns
- prelude::ops::einsum::utils::batch_transpose
- prelude::ops::einsum::utils::cache_friendly_trace
- prelude::ops::einsum::utils::compute_outer_product
- prelude::ops::elu
- prelude::ops::eq
- prelude::ops::erf
- prelude::ops::erfc
- prelude::ops::execute_fused_graph
- prelude::ops::exp
- prelude::ops::exp2
- prelude::ops::expand_as
- prelude::ops::expand_dims
- prelude::ops::expit
- prelude::ops::expm1
- prelude::ops::eye
- prelude::ops::fft
- prelude::ops::fft2
- prelude::ops::fft2_bf16
- prelude::ops::fft2_f16
- prelude::ops::fft2_inplace
- prelude::ops::fft3
- prelude::ops::fft3_inplace
- prelude::ops::fft::fft
- prelude::ops::fft::fft1d::fft
- prelude::ops::fft::fft1d::ifft
- prelude::ops::fft::fft1d::rfft
- prelude::ops::fft::fft2
- prelude::ops::fft::fft2_bf16
- prelude::ops::fft::fft2_f16
- prelude::ops::fft::fft2_inplace
- prelude::ops::fft::fft2d::fft2
- prelude::ops::fft::fft2d::ifft2
- prelude::ops::fft::fft3
- prelude::ops::fft::fft3_inplace
- prelude::ops::fft::fft3d::fft3
- prelude::ops::fft::fft3d::ifft3
- prelude::ops::fft::fft_bf16
- prelude::ops::fft::fft_f16
- prelude::ops::fft::fft_inplace
- prelude::ops::fft::generate_twiddle_factors
- prelude::ops::fft::gpu_kernels::gpu_fft2_dispatch
- prelude::ops::fft::gpu_kernels::gpu_fft3_dispatch
- prelude::ops::fft::gpu_kernels::gpu_fft_dispatch
- prelude::ops::fft::gpu_kernels::gpu_ifft2_dispatch
- prelude::ops::fft::gpu_kernels::gpu_ifft3_dispatch
- prelude::ops::fft::gpu_kernels::gpu_ifft_dispatch
- prelude::ops::fft::gpu_kernels::gpu_rfft_dispatch
- prelude::ops::fft::half_precision::fft2_bf16
- prelude::ops::fft::half_precision::fft2_f16
- prelude::ops::fft::half_precision::fft_bf16
- prelude::ops::fft::half_precision::fft_f16
- prelude::ops::fft::half_precision::ifft2_bf16
- prelude::ops::fft::half_precision::ifft2_f16
- prelude::ops::fft::half_precision::ifft_bf16
- prelude::ops::fft::half_precision::ifft_f16
- prelude::ops::fft::ifft
- prelude::ops::fft::ifft2
- prelude::ops::fft::ifft2_bf16
- prelude::ops::fft::ifft2_f16
- prelude::ops::fft::ifft2_inplace
- prelude::ops::fft::ifft3
- prelude::ops::fft::ifft3_inplace
- prelude::ops::fft::ifft_bf16
- prelude::ops::fft::ifft_f16
- prelude::ops::fft::ifft_inplace
- prelude::ops::fft::inplace::fft2_inplace
- prelude::ops::fft::inplace::fft3_inplace
- prelude::ops::fft::inplace::fft_inplace
- prelude::ops::fft::inplace::ifft2_inplace
- prelude::ops::fft::inplace::ifft3_inplace
- prelude::ops::fft::inplace::ifft_inplace
- prelude::ops::fft::rfft
- prelude::ops::fft::utils::generate_bit_reversal_table
- prelude::ops::fft::utils::generate_twiddle_factors
- prelude::ops::fft_bf16
- prelude::ops::fft_f16
- prelude::ops::fft_inplace
- prelude::ops::fix
- prelude::ops::flatten
- prelude::ops::flip
- prelude::ops::floor
- prelude::ops::fmax
- prelude::ops::fmin
- prelude::ops::fmod
- prelude::ops::fractional_avg_pool2d
- prelude::ops::fractional_max_pool2d
- prelude::ops::framework_comparison::print_framework_comparison_results
- prelude::ops::framework_comparison::run_framework_comparison_benchmark
- prelude::ops::from_range
- prelude::ops::fromfunction
- prelude::ops::full
- prelude::ops::fusion::execute_fused_graph
- prelude::ops::fusion::get_fusion_stats
- prelude::ops::fusion::print_fusion_report
- prelude::ops::fusion::record_fusion_opportunity
- prelude::ops::fusion::reset_fusion_stats
- prelude::ops::gamma
- prelude::ops::gather
- prelude::ops::ge
- prelude::ops::gelu
- prelude::ops::gelu_tanh
- prelude::ops::geomspace
- prelude::ops::get_fusion_stats
- prelude::ops::get_performance_analytics
- prelude::ops::global_avg_pool2d
- prelude::ops::global_avg_pool3d
- prelude::ops::global_max_pool2d
- prelude::ops::global_max_pool3d
- prelude::ops::gpu_tensorflow_benchmark::run_quick_gpu_tensorflow_benchmark
- prelude::ops::group_norm
- prelude::ops::gt
- prelude::ops::hard_sigmoid
- prelude::ops::hard_swish
- prelude::ops::hardswish
- prelude::ops::histogram
- prelude::ops::identity
- prelude::ops::ifft
- prelude::ops::ifft2
- prelude::ops::ifft2_bf16
- prelude::ops::ifft2_f16
- prelude::ops::ifft2_inplace
- prelude::ops::ifft3
- prelude::ops::ifft3_inplace
- prelude::ops::ifft_bf16
- prelude::ops::ifft_f16
- prelude::ops::ifft_inplace
- prelude::ops::infer_binary_elementwise
- prelude::ops::infer_binary_elementwise_validated
- prelude::ops::infer_concat
- prelude::ops::infer_conv2d
- prelude::ops::infer_matmul
- prelude::ops::infer_reduction
- prelude::ops::infer_reshape
- prelude::ops::inv
- prelude::ops::inverse_lapack
- prelude::ops::is_async_operations_idle
- prelude::ops::is_lapack_available
- prelude::ops::isfinite
- prelude::ops::isinf
- prelude::ops::isnan
- prelude::ops::kurtosis
- prelude::ops::lapack::cholesky_lapack
- prelude::ops::lapack::determinant_lapack
- prelude::ops::lapack::eigenvalues_lapack
- prelude::ops::lapack::inverse_lapack
- prelude::ops::lapack::is_lapack_available
- prelude::ops::lapack::lapack_provider
- prelude::ops::lapack::lstsq_lapack
- prelude::ops::lapack::lu_decompose_lapack
- prelude::ops::lapack::matmul_blas
- prelude::ops::lapack::pinv
- prelude::ops::lapack::qr_lapack
- prelude::ops::lapack::solve_lapack
- prelude::ops::lapack::svd_lapack
- prelude::ops::lapack_f32::determinant_f32
- prelude::ops::lapack_f32::inverse_f32
- prelude::ops::lapack_f32::solve_f32
- prelude::ops::lapack_f32::svd_f32
- prelude::ops::lapack_f64::determinant_f64
- prelude::ops::lapack_f64::inverse_f64
- prelude::ops::lapack_f64::solve_f64
- prelude::ops::lapack_f64::svd_f64
- prelude::ops::lapack_provider
- prelude::ops::layer_norm
- prelude::ops::le
- prelude::ops::leaky_relu
- prelude::ops::lgamma
- prelude::ops::linalg::cholesky
- prelude::ops::linalg::det
- prelude::ops::linalg::eig
- prelude::ops::linalg::inv
- prelude::ops::linalg::lu
- prelude::ops::linalg::qr_decomposition
- prelude::ops::linalg::svd
- prelude::ops::linspace
- prelude::ops::list_ufuncs
- prelude::ops::log
- prelude::ops::log10
- prelude::ops::log1p
- prelude::ops::log2
- prelude::ops::log_sigmoid
- prelude::ops::log_softmax
- prelude::ops::logical::logical_and
- prelude::ops::logical::logical_binary_op
- prelude::ops::logical::logical_not
- prelude::ops::logical::logical_or
- prelude::ops::logical::logical_unary_op
- prelude::ops::logical::logical_xor
- prelude::ops::logical_and
- prelude::ops::logical_not
- prelude::ops::logical_or
- prelude::ops::logical_xor
- prelude::ops::logit
- prelude::ops::logspace
- prelude::ops::logsumexp
- prelude::ops::lstsq_lapack
- prelude::ops::lt
- prelude::ops::lu
- prelude::ops::lu_decompose_lapack
- prelude::ops::manipulation::broadcast_indices
- prelude::ops::manipulation::broadcast_to
- prelude::ops::manipulation::calculate_strides
- prelude::ops::manipulation::cast
- prelude::ops::manipulation::common::broadcast_indices
- prelude::ops::manipulation::common::calculate_strides
- prelude::ops::manipulation::common::coords_to_flat
- prelude::ops::manipulation::common::flat_to_coords
- prelude::ops::manipulation::concat
- prelude::ops::manipulation::concatenation::concat
- prelude::ops::manipulation::concatenation::expand_dims
- prelude::ops::manipulation::concatenation::repeat
- prelude::ops::manipulation::concatenation::slice
- prelude::ops::manipulation::concatenation::split
- prelude::ops::manipulation::concatenation::stack
- prelude::ops::manipulation::concatenation::tile
- prelude::ops::manipulation::coords_to_flat
- prelude::ops::manipulation::expand_as
- prelude::ops::manipulation::expand_dims
- prelude::ops::manipulation::flat_to_coords
- prelude::ops::manipulation::flatten
- prelude::ops::manipulation::flip
- prelude::ops::manipulation::gather
- prelude::ops::manipulation::identity
- prelude::ops::manipulation::indexing::gather
- prelude::ops::manipulation::indexing::scatter
- prelude::ops::manipulation::indexing::select
- prelude::ops::manipulation::indexing::slice
- prelude::ops::manipulation::indexing::slice_with_stride
- prelude::ops::manipulation::indexing::where_op
- prelude::ops::manipulation::one_hot
- prelude::ops::manipulation::pad
- prelude::ops::manipulation::repeat
- prelude::ops::manipulation::reshape
- prelude::ops::manipulation::roll
- prelude::ops::manipulation::scatter
- prelude::ops::manipulation::select
- prelude::ops::manipulation::shape::broadcast_to
- prelude::ops::manipulation::shape::expand_as
- prelude::ops::manipulation::shape::expand_dims
- prelude::ops::manipulation::shape::flatten
- prelude::ops::manipulation::shape::reshape
- prelude::ops::manipulation::shape::squeeze
- prelude::ops::manipulation::shape::unsqueeze
- prelude::ops::manipulation::slice
- prelude::ops::manipulation::slice_with_stride
- prelude::ops::manipulation::split
- prelude::ops::manipulation::squeeze
- prelude::ops::manipulation::stack
- prelude::ops::manipulation::tile
- prelude::ops::manipulation::transpose
- prelude::ops::manipulation::transpose::flip
- prelude::ops::manipulation::transpose::roll
- prelude::ops::manipulation::transpose::transpose
- prelude::ops::manipulation::transpose::transpose_axes
- prelude::ops::manipulation::transpose_axes
- prelude::ops::manipulation::unsqueeze
- prelude::ops::manipulation::utilities::cast
- prelude::ops::manipulation::utilities::identity
- prelude::ops::manipulation::utilities::one_hot
- prelude::ops::manipulation::utilities::pad
- prelude::ops::manipulation::where_op
- prelude::ops::matmul
- prelude::ops::matmul::batch::extract_2d_slice
- prelude::ops::matmul::batch::matmul_batch
- prelude::ops::matmul::batch::store_2d_slice
- prelude::ops::matmul::batch_matmul
- prelude::ops::matmul::core::batch_matmul
- prelude::ops::matmul::core::dot
- prelude::ops::matmul::core::matmul
- prelude::ops::matmul::dot
- prelude::ops::matmul::gpu::matmul_batch_gpu
- prelude::ops::matmul::gpu::matmul_gpu_2d
- prelude::ops::matmul::gpu::matmul_mixed_precision_gpu
- prelude::ops::matmul::matmul
- prelude::ops::matmul::matmul_mixed_precision
- prelude::ops::matmul::optimized::matmul_2d_optimized
- prelude::ops::matmul::optimized::matmul_blas_f32
- prelude::ops::matmul::optimized::matmul_blas_f64
- prelude::ops::matmul::optimized::matmul_blocked
- prelude::ops::matmul::optimized::matmul_blocked_optimized
- prelude::ops::matmul::optimized::matmul_cache_optimized
- prelude::ops::matmul::optimized::matmul_simple_optimized
- prelude::ops::matmul::outer
- prelude::ops::matmul::shapes::broadcast_shapes
- prelude::ops::matmul::shapes::compute_broadcast_indices
- prelude::ops::matmul::shapes::compute_matmul_shape
- prelude::ops::matmul::specialized::matmul_mixed_precision
- prelude::ops::matmul::specialized::matmul_outer_product
- prelude::ops::matmul::specialized::outer
- prelude::ops::matmul_blas
- prelude::ops::max
- prelude::ops::max_pool2d
- prelude::ops::max_pool3d
- prelude::ops::maximum
- prelude::ops::mean
- prelude::ops::median
- prelude::ops::meshgrid
- prelude::ops::min
- prelude::ops::minimum
- prelude::ops::mish
- prelude::ops::modulo
- prelude::ops::moment
- prelude::ops::mul
- prelude::ops::mul_async
- prelude::ops::mul_async_priority
- prelude::ops::multinomial_f32
- prelude::ops::ne
- prelude::ops::neg
- prelude::ops::negative
- prelude::ops::normalization::batch_norm
- prelude::ops::normalization::batch_norm::batch_norm
- prelude::ops::normalization::group_norm
- prelude::ops::normalization::group_norm::group_norm
- prelude::ops::normalization::layer_norm
- prelude::ops::normalization::layer_norm::layer_norm
- prelude::ops::normalization::sync_batch_norm
- prelude::ops::normalization::sync_batch_norm::sync_batch_norm
- prelude::ops::np_identity
- prelude::ops::np_tanh
- prelude::ops::numpy_array_creation::arange
- prelude::ops::numpy_array_creation::diag
- prelude::ops::numpy_array_creation::diagonal
- prelude::ops::numpy_array_creation::eye
- prelude::ops::numpy_array_creation::from_range
- prelude::ops::numpy_array_creation::fromfunction
- prelude::ops::numpy_array_creation::full
- prelude::ops::numpy_array_creation::geomspace
- prelude::ops::numpy_array_creation::identity
- prelude::ops::numpy_array_creation::linspace
- prelude::ops::numpy_array_creation::logspace
- prelude::ops::numpy_array_creation::meshgrid
- prelude::ops::numpy_array_creation::ones
- prelude::ops::numpy_array_creation::tri
- prelude::ops::numpy_array_creation::tril
- prelude::ops::numpy_array_creation::triu
- prelude::ops::numpy_array_creation::zeros
- prelude::ops::numpy_broadcast_arrays
- prelude::ops::numpy_compat::absolute
- prelude::ops::numpy_compat::apply_ufunc
- prelude::ops::numpy_compat::arccos
- prelude::ops::numpy_compat::arccosh
- prelude::ops::numpy_compat::arcsin
- prelude::ops::numpy_compat::arcsinh
- prelude::ops::numpy_compat::arctan
- prelude::ops::numpy_compat::arctan2
- prelude::ops::numpy_compat::arctanh
- prelude::ops::numpy_compat::ceil
- prelude::ops::numpy_compat::cos
- prelude::ops::numpy_compat::cosh
- prelude::ops::numpy_compat::create_f32_ufunc_registry
- prelude::ops::numpy_compat::exp
- prelude::ops::numpy_compat::exp2
- prelude::ops::numpy_compat::expm1
- prelude::ops::numpy_compat::fix
- prelude::ops::numpy_compat::floor
- prelude::ops::numpy_compat::fmax
- prelude::ops::numpy_compat::fmin
- prelude::ops::numpy_compat::fmod
- prelude::ops::numpy_compat::isfinite
- prelude::ops::numpy_compat::isinf
- prelude::ops::numpy_compat::isnan
- prelude::ops::numpy_compat::list_ufuncs
- prelude::ops::numpy_compat::log
- prelude::ops::numpy_compat::log10
- prelude::ops::numpy_compat::log1p
- prelude::ops::numpy_compat::log2
- prelude::ops::numpy_compat::maximum
- prelude::ops::numpy_compat::minimum
- prelude::ops::numpy_compat::modulo
- prelude::ops::numpy_compat::negative
- prelude::ops::numpy_compat::numpy_broadcast_arrays
- prelude::ops::numpy_compat::reciprocal
- prelude::ops::numpy_compat::remainder
- prelude::ops::numpy_compat::rint
- prelude::ops::numpy_compat::sign
- prelude::ops::numpy_compat::signbit
- prelude::ops::numpy_compat::sin
- prelude::ops::numpy_compat::sinh
- prelude::ops::numpy_compat::sqrt
- prelude::ops::numpy_compat::square
- prelude::ops::numpy_compat::tan
- prelude::ops::numpy_compat::tanh
- prelude::ops::numpy_compat::trunc
- prelude::ops::one_hot
- prelude::ops::ones
- prelude::ops::optimized_binary::optimized_add
- prelude::ops::optimized_binary::optimized_binary_op
- prelude::ops::optimized_binary::optimized_binary_op_copy
- prelude::ops::optimized_binary::optimized_div
- prelude::ops::optimized_binary::optimized_mul
- prelude::ops::optimized_binary::optimized_sub
- prelude::ops::optimized_matmul_v2::ultra_matmul_v2
- prelude::ops::outer
- prelude::ops::pad
- prelude::ops::percentile
- prelude::ops::performance_benchmark::benchmark_binary_operations
- prelude::ops::performance_benchmark::print_benchmark_results
- prelude::ops::performance_benchmark::run_performance_benchmark
- prelude::ops::pooling::adaptive_avg_pool2d
- prelude::ops::pooling::adaptive_max_pool2d
- prelude::ops::pooling::adaptive_pooling::adaptive_avg_pool2d
- prelude::ops::pooling::adaptive_pooling::adaptive_max_pool2d
- prelude::ops::pooling::avg_pool2d
- prelude::ops::pooling::avg_pool3d
- prelude::ops::pooling::basic_pooling::avg_pool2d
- prelude::ops::pooling::basic_pooling::avg_pool3d
- prelude::ops::pooling::basic_pooling::max_pool2d
- prelude::ops::pooling::basic_pooling::max_pool3d
- prelude::ops::pooling::fractional_avg_pool2d
- prelude::ops::pooling::fractional_max_pool2d
- prelude::ops::pooling::fractional_pooling::fractional_avg_pool2d
- prelude::ops::pooling::fractional_pooling::fractional_max_pool2d
- prelude::ops::pooling::global_avg_pool2d
- prelude::ops::pooling::global_avg_pool3d
- prelude::ops::pooling::global_max_pool2d
- prelude::ops::pooling::global_max_pool3d
- prelude::ops::pooling::global_pooling::global_avg_pool2d
- prelude::ops::pooling::global_pooling::global_avg_pool3d
- prelude::ops::pooling::global_pooling::global_max_pool2d
- prelude::ops::pooling::global_pooling::global_max_pool3d
- prelude::ops::pooling::max_pool2d
- prelude::ops::pooling::max_pool3d
- prelude::ops::pooling::roi_align2d
- prelude::ops::pooling::roi_pool2d
- prelude::ops::pooling::roi_pooling::roi_align2d
- prelude::ops::pooling::roi_pooling::roi_pool2d
- prelude::ops::pow
- prelude::ops::pow_async
- prelude::ops::prelu
- prelude::ops::prelu_async
- prelude::ops::print_framework_comparison_results
- prelude::ops::print_fusion_report
- prelude::ops::qr_lapack
- prelude::ops::quantile
- prelude::ops::rand_f32
- prelude::ops::rand_f64
- prelude::ops::randn_f32
- prelude::ops::randn_f64
- prelude::ops::random::multinomial_f32
- prelude::ops::random::rand_f32
- prelude::ops::random::rand_f64
- prelude::ops::random::randn_f32
- prelude::ops::random::randn_f64
- prelude::ops::random::random_normal_f32
- prelude::ops::random::random_normal_f32_device
- prelude::ops::random::random_normal_f64
- prelude::ops::random::random_uniform_f32
- prelude::ops::random::random_uniform_f32_device
- prelude::ops::random::random_uniform_f64
- prelude::ops::random::random_uniform_int
- prelude::ops::random_normal_f32
- prelude::ops::random_normal_f32_device
- prelude::ops::random_normal_f64
- prelude::ops::random_uniform_f32
- prelude::ops::random_uniform_f32_device
- prelude::ops::random_uniform_f64
- prelude::ops::random_uniform_int
- prelude::ops::range
- prelude::ops::reciprocal
- prelude::ops::record_fusion_opportunity
- prelude::ops::reduction::all
- prelude::ops::reduction::any
- prelude::ops::reduction::argmax
- prelude::ops::reduction::argmin
- prelude::ops::reduction::argops::argmax
- prelude::ops::reduction::argops::argmin
- prelude::ops::reduction::argops::topk
- prelude::ops::reduction::boolean::all
- prelude::ops::reduction::boolean::any
- prelude::ops::reduction::common::normalize_axis
- prelude::ops::reduction::cumprod
- prelude::ops::reduction::cumsum
- prelude::ops::reduction::cumulative::cumprod
- prelude::ops::reduction::cumulative::cumsum
- prelude::ops::reduction::gpu_execution::execute_gpu_max_reduction
- prelude::ops::reduction::gpu_execution::execute_gpu_mean_reduction
- prelude::ops::reduction::gpu_execution::execute_gpu_min_reduction
- prelude::ops::reduction::gpu_execution::execute_gpu_product_reduction
- prelude::ops::reduction::gpu_execution::execute_gpu_reduction
- prelude::ops::reduction::gpu_execution::execute_gpu_sum_reduction
- prelude::ops::reduction::gpu_execution::execute_gpu_variance_reduction
- prelude::ops::reduction::gpu_kernels::gpu_l1_norm_axis
- prelude::ops::reduction::gpu_kernels::gpu_l2_norm_axis
- prelude::ops::reduction::gpu_kernels::gpu_max_axis
- prelude::ops::reduction::gpu_kernels::gpu_mean_axis
- prelude::ops::reduction::gpu_kernels::gpu_min_axis
- prelude::ops::reduction::gpu_kernels::gpu_reduce_all
- prelude::ops::reduction::gpu_kernels::gpu_reduce_axis
- prelude::ops::reduction::gpu_kernels::gpu_std_axis
- prelude::ops::reduction::gpu_kernels::gpu_sum_axis
- prelude::ops::reduction::gpu_kernels::gpu_var_axis
- prelude::ops::reduction::max
- prelude::ops::reduction::mean
- prelude::ops::reduction::min
- prelude::ops::reduction::normalize_axis
- prelude::ops::reduction::prod
- prelude::ops::reduction::segment::minmax::segment_max
- prelude::ops::reduction::segment::minmax::segment_min
- prelude::ops::reduction::segment::prod_any_all::segment_all
- prelude::ops::reduction::segment::prod_any_all::segment_any
- prelude::ops::reduction::segment::prod_any_all::segment_prod
- prelude::ops::reduction::segment::segment_all
- prelude::ops::reduction::segment::segment_any
- prelude::ops::reduction::segment::segment_max
- prelude::ops::reduction::segment::segment_mean
- prelude::ops::reduction::segment::segment_min
- prelude::ops::reduction::segment::segment_prod
- prelude::ops::reduction::segment::segment_sum
- prelude::ops::reduction::segment::sum_mean::segment_mean
- prelude::ops::reduction::segment::sum_mean::segment_sum
- prelude::ops::reduction::segment::unsorted::unsorted_segment_max
- prelude::ops::reduction::segment::unsorted::unsorted_segment_mean
- prelude::ops::reduction::segment::unsorted::unsorted_segment_min
- prelude::ops::reduction::segment::unsorted::unsorted_segment_prod
- prelude::ops::reduction::segment::unsorted::unsorted_segment_sum
- prelude::ops::reduction::segment::unsorted_segment_max
- prelude::ops::reduction::segment::unsorted_segment_mean
- prelude::ops::reduction::segment::unsorted_segment_min
- prelude::ops::reduction::segment::unsorted_segment_prod
- prelude::ops::reduction::segment::unsorted_segment_sum
- prelude::ops::reduction::segment_max
- prelude::ops::reduction::segment_mean
- prelude::ops::reduction::segment_sum
- prelude::ops::reduction::statistical::max
- prelude::ops::reduction::statistical::mean
- prelude::ops::reduction::statistical::min
- prelude::ops::reduction::statistical::prod
- prelude::ops::reduction::statistical::reduce_all_cpu
- prelude::ops::reduction::statistical::reduce_axis_cpu
- prelude::ops::reduction::statistical::sum
- prelude::ops::reduction::statistical::variance
- prelude::ops::reduction::sum
- prelude::ops::reduction::topk
- prelude::ops::reduction::variance
- prelude::ops::registry_extensions::generate_performance_report
- prelude::ops::registry_extensions::get_kernel_smart
- prelude::ops::registry_extensions::print_performance_report
- prelude::ops::registry_extensions::warm_kernels
- prelude::ops::relu
- prelude::ops::relu6
- prelude::ops::remainder
- prelude::ops::repeat
- prelude::ops::reset_fusion_stats
- prelude::ops::reshape
- prelude::ops::rfft
- prelude::ops::rint
- prelude::ops::roi_align2d
- prelude::ops::roi_pool2d
- prelude::ops::roll
- prelude::ops::run_framework_comparison_benchmark
- prelude::ops::run_quick_gpu_tensorflow_benchmark
- prelude::ops::scalar_add
- prelude::ops::scatter
- prelude::ops::select
- prelude::ops::selu
- prelude::ops::shape_inference::infer_binary_elementwise
- prelude::ops::shape_inference::infer_binary_elementwise_validated
- prelude::ops::shape_inference::infer_concat
- prelude::ops::shape_inference::infer_conv2d
- prelude::ops::shape_inference::infer_matmul
- prelude::ops::shape_inference::infer_reduction
- prelude::ops::shape_inference::infer_reshape
- prelude::ops::shape_inference_registry::get_registry
- prelude::ops::shape_inference_registry::global::get_registry
- prelude::ops::shape_inference_registry::global::initialize_registry
- prelude::ops::shape_inference_registry::initialize_registry
- prelude::ops::sigmoid
- prelude::ops::sign
- prelude::ops::signbit
- prelude::ops::sin
- prelude::ops::sinh
- prelude::ops::skewness
- prelude::ops::slice
- prelude::ops::slice_with_stride
- prelude::ops::smooth_l1_loss
- prelude::ops::softmax
- prelude::ops::softplus
- prelude::ops::softsign
- prelude::ops::solve_lapack
- prelude::ops::special::bessel_j0
- prelude::ops::special::bessel_j1
- prelude::ops::special::bessel_y0
- prelude::ops::special::bessel_y1
- prelude::ops::special::digamma
- prelude::ops::special::erf
- prelude::ops::special::erfc
- prelude::ops::special::gamma
- prelude::ops::special::lgamma
- prelude::ops::special::smooth_l1_loss
- prelude::ops::split
- prelude::ops::sqrt
- prelude::ops::square
- prelude::ops::squeeze
- prelude::ops::stack
- prelude::ops::stats::clear_performance_metrics
- prelude::ops::stats::config::clear_performance_metrics
- prelude::ops::stats::config::generate_performance_report
- prelude::ops::stats::config::get_performance_metrics
- prelude::ops::stats::config::get_stats_config
- prelude::ops::stats::config::set_stats_config
- prelude::ops::stats::correlation
- prelude::ops::stats::covariance
- prelude::ops::stats::distribution::correlation
- prelude::ops::stats::distribution::covariance
- prelude::ops::stats::distribution::median
- prelude::ops::stats::distribution::quantile
- prelude::ops::stats::generate_performance_report
- prelude::ops::stats::get_performance_metrics
- prelude::ops::stats::get_stats_config
- prelude::ops::stats::histogram
- prelude::ops::stats::histogram::histogram
- prelude::ops::stats::kurtosis
- prelude::ops::stats::median
- prelude::ops::stats::moment
- prelude::ops::stats::moments::kurtosis
- prelude::ops::stats::moments::moment
- prelude::ops::stats::moments::percentile
- prelude::ops::stats::moments::range
- prelude::ops::stats::moments::skewness
- prelude::ops::stats::percentile
- prelude::ops::stats::quantile
- prelude::ops::stats::range
- prelude::ops::stats::set_stats_config
- prelude::ops::stats::skewness
- prelude::ops::sub
- prelude::ops::sub_async
- prelude::ops::sum
- prelude::ops::svd
- prelude::ops::svd_lapack
- prelude::ops::swish
- prelude::ops::synchronize_async_operations
- prelude::ops::tan
- prelude::ops::tanh
- prelude::ops::tile
- prelude::ops::topk
- prelude::ops::transpose
- prelude::ops::tri
- prelude::ops::tril
- prelude::ops::triu
- prelude::ops::trunc
- prelude::ops::ultra_matmul
- prelude::ops::ultra_matmul_v2
- prelude::ops::ultra_matmul_v3
- prelude::ops::ultra_matmul_v3::clear_performance_analytics
- prelude::ops::ultra_matmul_v3::configure_ultra_performance
- prelude::ops::ultra_matmul_v3::get_performance_analytics
- prelude::ops::ultra_matmul_v3::ultra_matmul_v3
- prelude::ops::ultra_performance_matmul::ultra_matmul
- prelude::ops::unary::abs
- prelude::ops::unary::absolute
- prelude::ops::unary::ceil
- prelude::ops::unary::cos
- prelude::ops::unary::cosh
- prelude::ops::unary::exp
- prelude::ops::unary::exp2
- prelude::ops::unary::expm1
- prelude::ops::unary::fix
- prelude::ops::unary::floor
- prelude::ops::unary::log
- prelude::ops::unary::log10
- prelude::ops::unary::log1p
- prelude::ops::unary::log2
- prelude::ops::unary::neg
- prelude::ops::unary::negative
- prelude::ops::unary::reciprocal
- prelude::ops::unary::rint
- prelude::ops::unary::sign
- prelude::ops::unary::signbit
- prelude::ops::unary::sin
- prelude::ops::unary::sinh
- prelude::ops::unary::sqrt
- prelude::ops::unary::square
- prelude::ops::unary::tan
- prelude::ops::unary::tanh
- prelude::ops::unary::trunc
- prelude::ops::unified_dispatch::dispatch_op
- prelude::ops::unsqueeze
- prelude::ops::where_op
- prelude::ops::zeros
- prelude::quick_train::quick_experiment
- prelude::quick_train::train_with_adam
- prelude::quick_train::train_with_sgd
- utils::bytes_to_human_readable
- utils::clamp
- utils::clamp_slice_inplace
- utils::flat_index
- utils::lerp
- utils::log_softmax
- utils::next_power_of_two
- utils::num_elements
- utils::shape_to_strides
- utils::sigmoid
- utils::sigmoid_slice
- utils::softmax
- version
- version_check::assert_versions_consistent
- version_check::check_version_consistency
- version_check::subcrate_versions
- version_info
Type Aliases
- CoreResult
- common::Shape
- error::Result
- logging::LogBackend
- prelude::CoreResult
- prelude::ops::registry::ShapeFn
- prelude::ops::registry::types::ShapeFn
- prelude::ops::shape_inference_registry::OperationMetadata
- prelude::ops::shape_inference_registry::ShapeInferenceFn
- prelude::ops::shape_inference_registry::types::OperationMetadata
- prelude::ops::shape_inference_registry::types::ShapeInferenceFn
- type_aliases::BatchTensor
- type_aliases::Matrix
- type_aliases::Tensor1D
- type_aliases::Tensor1Df32
- type_aliases::Tensor1Df64
- type_aliases::Tensor2D
- type_aliases::Tensor2Df32
- type_aliases::Tensor2Df64
- type_aliases::Tensor3D
- type_aliases::Tensor3Df32
- type_aliases::Tensor3Df64
- type_aliases::Tensor4D
- type_aliases::Tensor4Df32
- type_aliases::Tensor4Df64
- type_aliases::Vector
Constants
- VERSION
- prelude::ops::activation::APPROX_THRESHOLD
- prelude::ops::activation::GPU_THRESHOLD
- prelude::ops::activation::PARALLEL_THRESHOLD
- prelude::ops::activation::SIMD_THRESHOLD
- prelude::ops::activation::strategy::APPROX_THRESHOLD
- prelude::ops::activation::strategy::GPU_THRESHOLD
- prelude::ops::activation::strategy::PARALLEL_THRESHOLD
- prelude::ops::activation::strategy::SIMD_THRESHOLD