MCPcopy Create free account

hub / github.com/intel/neural-compressor / endpoints

Endpoints36 in github.com/intel/neural-compressor

⨍ Functions 4,695◇ Types & classes 839↳ Endpoints 36
Routedo_transformation
dump_elapsed_time("Pass PostHostConstConverter")
neural_compressor/tensorflow/quantization/utils/graph_rewriter/int8/post_hostconst_converter.py:None
Routedo_transformation
dump_elapsed_time("Pass PostCseOptimizer")
neural_compressor/tensorflow/quantization/utils/graph_rewriter/int8/post_quantized_op_cse.py:None
Routedo_transformation
dump_elapsed_time("Pass SplitSharedInputOptimizer")
neural_compressor/tensorflow/quantization/utils/graph_rewriter/generic/split_shared_input.py:None
Routefloat_get_scale
partial(jax.lax.composite, name="inc.get_scale_fp8")
neural_compressor/jax/utils/utility.py:None
Routeinput_tensor_names
input_tensor_names.setter
neural_compressor/tensorflow/utils/model_wrappers.py:None
Routeinteger_get_q_params
partial(jax.lax.composite, name="inc.get_q_params_int")
neural_compressor/jax/utils/utility.py:None
Routeinteger_get_scale
partial(jax.lax.composite, name="inc.get_scale_int")
neural_compressor/jax/utils/utility.py:None
Routeoutput_tensor_names
output_tensor_names.setter
neural_compressor/tensorflow/utils/model_wrappers.py:None
Routetest_autotune_api
reset_tuning_target
test/torch/test_autotune.py:None
Routetest_autotune_api_2
reset_tuning_target
test/torch/test_autotune.py:None
Routetest_autotune_get_config_set_api
reset_tuning_target
test/torch/test_autotune.py:None
Routetest_autotune_mixed_precision_default
reset_tuning_target
test/torch/test_autotune.py:None
Routetest_autotune_mixed_precision_set_op_name
reset_tuning_target
test/torch/test_autotune.py:None
Routetest_do_transform
unittest.skipIf( version1_gte_version2(tf.version.VERSION, "2.16.1"), "The TF BN is deleted af
test/tensorflow/quantization/ptq/test_fold_batch_norm.py:None
Routetest_export_model_failed
patch("neural_compressor.torch.algorithms.pt2e_quant.core.logger.error")
test/torch/algorithms/pt2e_quant/test_pt2e_w8a8.py:None
Routetest_fold_output_values
unittest.skipIf( version1_gte_version2(tf.version.VERSION, "2.16.1"), "The TF BN is deleted af
test/tensorflow/quantization/ptq/test_fold_batch_norm.py:None
Routetest_missing_torchao_fails_fast
pytest.mark.skipif(get_torch_version() < TORCH_VERSION_2_11_0, reason="Requires torch>=2.11")
test/torch/algorithms/pt2e_quant/test_pt2e_w8a8.py:None
Routetest_no_assert
pytest.mark.parametrize("dynamic_scale_patching", [True, False])
test/torch/algorithms/fp8_quant/unit_tests/test_runtime_scale_patching.py:None
Routetest_numba_not_installed
patch.object(inc_torch_env, "is_numba_available", lambda: False)
test/torch/utils/test_torch_utility.py:None
Routetest_prepare_and_convert_on_llm
pytest.mark.skipif(get_torch_version() <= TORCH_VERSION_2_2_2, reason="Requires torch>=2.3.0")
test/torch/quantization/test_pt2e_quant.py:None
Routetest_prepare_and_convert_on_simple_model
pytest.mark.skipif(get_torch_version() <= TORCH_VERSION_2_2_2, reason="Requires torch>=2.3.0")
test/torch/quantization/test_pt2e_quant.py:None
Routetest_pt2e_compat_uses_torchao_on_torch_2_11_plus
pytest.mark.skipif(get_torch_version() < TORCH_VERSION_2_11_0, reason="Requires torch>=2.11")
test/torch/quantization/test_pt2e_quant.py:None
Routetest_quantize_input
unittest.skipIf( tf.version.VERSION < "2.1.0", "Quantize input needs tensorflow 2.1.0 and newe
test/tensorflow/quantization/ptq/test_quantize_input.py:None
Routetest_quantize_simple_model
pytest.mark.skipif(get_torch_version() <= TORCH_VERSION_2_2_2, reason="Requires torch>=2.3.0")
test/torch/quantization/test_pt2e_quant.py:None
Routetest_quantizer_on_llm
pytest.mark.skipif(get_torch_version() <= TORCH_VERSION_2_2_2, reason="Requires torch>=2.3.0")
test/torch/algorithms/pt2e_quant/test_pt2e_w8a8.py:None
Routetest_quantizer_on_simple_model
pytest.mark.skipif(get_torch_version() <= TORCH_VERSION_2_2_2, reason="Requires torch>=2.3.0")
test/torch/algorithms/pt2e_quant/test_pt2e_w8a8.py:None
Routetest_quantizer_on_simple_model
pytest.mark.skipif( torch_utils.get_torch_version() <= torch_utils.TORCH_VERSION_2_2_2, reason="Requir
test/torch/algorithms/pt2e_quant/test_half_precision_rewriter.py:None
Routetest_rtn_double_quant_config_set
reset_tuning_target
test/torch/test_autotune.py:None
Routetest_rtn_double_quant_config_set2
reset_tuning_target
test/torch/test_autotune.py:None
Routetest_rtn_double_quant_config_set3
patch("neural_compressor.torch.utils.constants.DOUBLE_QUANT_CONFIGS", FAKE_DOUBLE_QUANT_CONFIGS)
test/torch/test_autotune.py:None
Routetest_target_bits
pytest.mark.skipif(Version(auto_round.__version__) < Version("0.9.0"), reason="target bits is not supported.")
test/torch/quantization/test_autoround_cpu.py:None
Routetest_target_bits_autotune
pytest.mark.skipif(Version(auto_round.__version__) < Version("0.9.0"), reason="target bits is not supported.")
test/torch/quantization/test_autoround_cpu.py:None
Routetest_tbb_installed_but_not_configured_right
patch.object(inc_torch_env, "_is_tbb_installed", lambda: True)
test/torch/utils/test_torch_utility.py:None
Routetest_tbb_not_installed
patch.object(inc_torch_env, "_is_tbb_installed", lambda: False)
test/torch/utils/test_torch_utility.py:None
Routetest_torchao_imports_for_torch_ge_2_11
pytest.mark.skipif(get_torch_version() < TORCH_VERSION_2_11_0, reason="Requires torch>=2.11")
test/torch/algorithms/pt2e_quant/test_pt2e_w8a8.py:None
Routetest_warning_once
patch.object(Logger, "warning")
test/common/test_logger.py:None