MCPcopy Create free account

hub / github.com/MegEngine/MegEngine / endpoints

Endpoints66 in github.com/MegEngine/MegEngine

⨍ Functions 33,912◇ Types & classes 7,186↳ Endpoints 66
RouteEmbossKernel
subgraph_fn( "EmbossKernel", dtype=dtype, device=device, nr_inputs=2, gopt_level=None, )
imperative/python/megengine/module/vision.py:None
RouteSharpenKernel
subgraph_fn( "SharpenKernel", dtype=dtype, device=device, nr_inputs=2, gopt_level=None, )
imperative/python/megengine/module/vision.py:None
Route_compute_mode
_compute_mode.setter
imperative/python/megengine/core/_config.py:None
Routebatch_norm_nd
subgraph_fn( "BatchNormNd", dtype=dtype, device=device, nr_inputs=4, i
imperative/python/test/unit/core/test_subgraph.py:None
Routebinary_cross_entropy
_reduce_output
imperative/python/megengine/functional/loss.py:None
Routecross_entropy
_reduce_output
imperative/python/megengine/functional/loss.py:None
Routeeinsum
subgraph_fn( "Einsum", dtype=dtype, device=device, nr_inputs=len(input_ndims), gopt_level=2 )
imperative/python/megengine/functional/einsum.py:None
Routeget_conv_execution_strategy
deprecated(version="1.3", reason="use get_execution_strategy() instead")
imperative/python/megengine/functional/debug_param.py:None
Routeget_var_shape_lower
register_lower_rule(mops.GetVarShape)
imperative/python/megengine/xla/rules/trivial.py:None
Routehinge_loss
_reduce_output
imperative/python/megengine/functional/loss.py:None
Routel1_loss
_reduce_output
imperative/python/megengine/functional/loss.py:None
Routelogsigmoid
subgraph_fn( "LogSigmoid", dtype=dtype, device=device, nr_inputs=1, ji
imperative/python/test/unit/functional/test_elemwise.py:None
Routemul
subgraph_fn( "Mul", dtype=dtype, device=device, nr_inputs=2, gopt_leve
imperative/python/test/unit/core/test_subgraph.py:None
Routerequires_grad
requires_grad.deleter
imperative/python/megengine/tensor.py:None
Routesoftplus
subgraph_fn( "Softplus", dtype=dtype, device=device, nr_inputs=1, jit_
imperative/python/test/unit/functional/test_elemwise.py:None
Routesquare_loss
_reduce_output
imperative/python/megengine/functional/loss.py:None
Routetest_BetaRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_ExponentialRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_GammaRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_MultinomialRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_NormalRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_PermutationRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_PoissonRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_ShuffleRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_UniformRNG
pytest.mark.skipif( get_device_count("xpu") <= 1, reason="xpu counts need > 1", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_advance_indexing
pytest.mark.parametrize("mode", ["get", "set", "inc"])
imperative/python/test/unit/utils/test_network_node.py:None
Routetest_batch_conv_bias
pytest.mark.skipif(get_device_count("gpu") > 0, reason="no int8 algorithm on cuda")
imperative/python/test/unit/functional/test_functional.py:None
Routetest_batch_mesh_indexing
pytest.mark.parametrize("mode", ["get", "set", "inc"])
imperative/python/test/unit/utils/test_network_node.py:None
Routetest_beta_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_broadcast_on_trace
pytest.mark.parametrize( "input_shape, target_shapes", [ ((3,), [(2, 1, 3), (1, 2, 3), (2, 2,
imperative/python/test/unit/functional/test_tensor.py:None
Routetest_collect_results
pytest.mark.parametrize("output_size", [10, 10000], ids=["small_size", "large_size"])
imperative/python/test/unit/distributed/test_distributed.py:None
Routetest_conv_bias
pytest.mark.skipif( get_device_count("gpu") > 0, reason="cuda does not support nchw int8" )
imperative/python/test/unit/quantization/test_op.py:None
Routetest_conv_bias
pytest.mark.skipif( get_device_count("gpu") > 0, reason="cuda does not support nchw int8" )
imperative/python/test/unit/functional/test_functional.py:None
Routetest_conv_transpose2d
pytest.mark.skipif( get_device_count("gpu") > 0 and get_cuda_compute_capability(0) < 61, reason="does
imperative/python/test/unit/quantization/test_op.py:None
Routetest_convbias
pytest.mark.skipif( get_device_count("gpu") > 0 and get_cuda_compute_capability(0) < 61, reason="does
imperative/python/test/unit/utils/test_network_node.py:None
Routetest_cpu_func
pytest.mark.skipif( get_device_count("gpu") > 0, reason="elem_add operator is only supported on CPU" )
imperative/python/test/unit/core/test_custom_op.py:None
Routetest_cuda_dataloader_parallel_worker_exception
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_dataloader.py:None
Routetest_cuda_dataloader_parallel_worker_with_cvaug
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_dataloader.py:None
Routetest_cuda_predataloader_parallel_worker_exception
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_dataloader.py:None
Routetest_custom_op
pytest.mark.skipif( get_device_count("gpu") < 1, reason="matmul scale operator is only supported on GPU" )
imperative/python/test/unit/core/test_custom_op.py:None
Routetest_dataloader_parallel_timeout
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_pre_dataloader.py:None
Routetest_dataloader_parallel_timeout
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_dataloader.py:None
Routetest_dataloader_parallel_worker_exception
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_pre_dataloader.py:None
Routetest_dataloader_parallel_worker_exception
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_dataloader.py:None
Routetest_dataloader_worker_signal_exception
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_dataloader.py:None
Routetest_emulate_forward_mode_with_reverse_mode
pytest.mark.parametrize("target", [F.cos, F.sin, lambda x: x * 2 + 1])
imperative/python/test/unit/autodiff/test_grad_manager.py:None
Routetest_exponential_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_gamma_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_gaussian_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_gpu_func
pytest.mark.skipif( get_device_count("gpu") < 1, reason="matmul scale operator is only supported on GPU" )
imperative/python/test/unit/core/test_custom_op.py:None
Routetest_lstm
pytest.mark.skipif(get_device_count("gpu") > 0, reason="no algorithm on cuda")
imperative/python/test/unit/module/test_rnn.py:None
Routetest_lstm_cell
pytest.mark.skipif(get_device_count("gpu") > 0, reason="no algorithm on cuda")
imperative/python/test/unit/module/test_rnn.py:None
Routetest_mesh_indexing
pytest.mark.parametrize("mode", ["get", "set", "inc"])
imperative/python/test/unit/utils/test_network_node.py:None
Routetest_multinomial_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_multinomial_op_without_replacement
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_permutation_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_poisson_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routetest_predataloader_parallel_worker_exception
pytest.mark.skipif( multiprocessing.get_start_method() != "fork", reason="the runtime error is only ra
imperative/python/test/unit/data/test_dataloader.py:None
Routetest_qat_batchmatmul_activation
pytest.mark.skipif(get_device_count("gpu") > 0, reason="no int8 algorithm on cuda")
imperative/python/test/unit/module/test_qat.py:None
Routetest_rnn
pytest.mark.skipif(get_device_count("gpu") > 0, reason="no algorithm on cuda")
imperative/python/test/unit/module/test_rnn.py:None
Routetest_rnn_cell
pytest.mark.skipif(get_device_count("gpu") > 0, reason="no algorithm on cuda")
imperative/python/test/unit/module/test_rnn.py:None
Routetest_subgraph
pytest.mark.parametrize("device", [get_default_device(), "cpux"])
imperative/python/test/unit/core/test_subgraph.py:None
Routetest_subtensor
pytest.mark.parametrize("mode", ["get", "set", "inc"])
imperative/python/test/unit/utils/test_network_node.py:None
Routetest_trace_mismatch
pytest.mark.parametrize( "normal_expr, mismatch_expr, reason", [ ("a + b + c", "a + b - c", "o
imperative/python/test/unit/jit/test_tracing.py:None
Routetest_uniform_op
pytest.mark.skipif( get_device_count("xpu") <= 2, reason="xpu counts need > 2", )
imperative/python/test/unit/random/test_rng.py:None
Routewraped_fn
functools.wraps(orig_getattr)
imperative/python/megengine/traced_module/traced_module.py:None