MCPcopy Create free account

hub / github.com/alibaba/BladeDISC / functions

Functions6,088 in github.com/alibaba/BladeDISC

↓ 1 callersFunction_attention
(query: torch.Tensor, key: torch.Tensor, value: torch.Tensor, biases: List[torch.Tensor])
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/ops.py:266
↓ 1 callersFunction_benchmark
()
pytorch_blade/torch_blade/testing/common_utils.py:152
↓ 1 callersFunction_broadcast_unsupported_set
(block, trt_unsupported, support_number_inpts_outs)
pytorch_blade/torch_blade/clustering/support_fusion_algorithm.py:254
↓ 1 callersFunction_build_group_support_info
(topo_nodes, trt_unsupported, graph_builder)
pytorch_blade/torch_blade/clustering/support_fusion_algorithm.py:161
↓ 1 callersFunction_build_onnx_engine
(subgraph, engine_build_func, group_name, dynamic_settings=None, cast_int_to_i32=False,
pytorch_blade/torch_blade/onnx_backends/backend_conversion.py:48
↓ 1 callersMethod_build_tensorrt_engine
( self, model_proto: onnx.ModelProto, disable_fp16: bool, shapes_min: List[Sha
tensorflow_blade/tf_blade/gpu/tf_to_trt.py:142
↓ 1 callersFunction_cal_diff
(origin_output, new_output)
pytorch_blade/torch_blade/tools/low_precision_analysis.py:49
↓ 1 callersFunction_calculate_fan
(linear_weight_shape, fan="fan_in")
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/ops.py:31
↓ 1 callersMethod_chain_relpos_emb
( self, rp, rp_entity_id, rp_rel_chain, )
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/embedder.py:94
↓ 1 callersMethod_check_concrete_shape
(self, graph)
pytorch_blade/torch_blade/onnx_backends/backend_testbed.py:30
↓ 1 callersMethod_check_disc_fixed_shape
(self, fn)
tao/tao_bridge/test/gpu/mlir/test_mlir_static_rank_cache.py:101
↓ 1 callersMethod_check_disc_time_const
(self, fn)
tao/tao_bridge/test/gpu/mlir/test_mlir_static_rank_cache.py:41
↓ 1 callersFunction_check_dynamic_ranges
(val)
pytorch_blade/torch_blade/config.py:44
↓ 1 callersMethod_check_tao_fixed_shape
(self, fn)
tao/tao_bridge/test/gpu/mlir/test_mlir_static_rank_cache.py:92
↓ 1 callersMethod_check_tao_time_const
(self, fn)
tao/tao_bridge/test/gpu/mlir/test_mlir_static_rank_cache.py:31
↓ 1 callersMethod_chunk
(self, z, mask, weight_ag, bias_ag, weight_ap, bias_ap, weight_bg, bias_bg, weight_bp, bias_bp, chunk_size)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/triangle.py:76
↓ 1 callersMethod_chunk
(self, x: torch.Tensor, biases: List[torch.Tensor], chunk_size: int, )
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/triangle.py:199
↓ 1 callersMethod_chunk
(self, m: torch.Tensor, mask: torch.Tensor, chunk_size: int, )
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/ops.py:229
↓ 1 callersMethod_chunk
(self, m: torch.Tensor, biases: List[torch.Tensor], chunk_size: int, )
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/msa.py:73
↓ 1 callersMethod_chunk
(self, m: torch.Tensor, mask: torch.Tensor, chunk_size: torch.Tensor, )
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/msa.py:212
↓ 1 callersMethod_chunk
(self, a: torch.Tensor, b: torch.Tensor, norm: torch.Tensor, chunk_size: in
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/outer_product_mean.py:76
↓ 1 callersMethod_chunk
(self, x, z)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/embedder.py:213
↓ 1 callersMethod_chunk
(self, z: torch.Tensor, t: torch.Tensor, biases: List[torch.Tensor], chunk_siz
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/template.py:77
↓ 1 callersMethod_chunk_relpos_emb
( self, rp, rp_entity_id, rp_rel_chain )
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/embedder.py:109
↓ 1 callersFunction_cluster_by_union_find
(graph_builder, support_info)
pytorch_blade/torch_blade/clustering/support_fusion_algorithm.py:175
↓ 1 callersFunction_collect_all_inplace_nodes
(block)
pytorch_blade/torch_blade/pass_manager.py:405
↓ 1 callersFunction_collect_all_tensors
(block)
pytorch_blade/torch_blade/tools/shape_inference.py:67
↓ 1 callersMethod_combine_histograms
( self, orig_hist: torch.Tensor, new_hist: torch.Tensor, upsam
tools/torch_quant/torch_quant/observer.py:591
↓ 1 callersFunction_compile_torchscript
(graph)
pytorch_blade/torch_blade/mlir/disc_engine_conversion.py:35
↓ 1 callersMethod_compute_quantization_error
r""" Compute the quantization error if we use start_bin to end_bin as the min and max to do the quantization.
tools/torch_quant/torch_quant/observer.py:452
↓ 1 callersMethod_conv_net
( self, x: tf.Tensor, weights: tf.Tensor, biases: tf.Tensor )
tensorflow_blade/tests/tf_test_common.py:35
↓ 1 callersMethod_convert_subgraph_to_onnx
( self, index: int, subgraph: tf.GraphDef, new_input_names: List[str],
tensorflow_blade/tf_blade/gpu/tf_to_trt.py:101
↓ 1 callersMethod_copy_graph_and_replace_input
Copy node from the original graph and replace the input names.
tensorflow_blade/tf_blade/util/simple_graph.py:567
↓ 1 callersFunction_createGraphByTracing
pytorch_blade/pytorch_blade/compiler/jit/pybind_functions.cpp:37
↓ 1 callersFunction_create_graph_builder
(non_const_topolist, support_number_inpts_outs, trt_unsupported)
pytorch_blade/torch_blade/clustering/support_fusion_algorithm.py:115
↓ 1 callersFunction_create_graph_def_ctl_depenency
Placeholder|px ──────► Abs|y1 ─────────────────┐ │ . │ │ .ctl_dep
tensorflow_blade/tests/util/simple_graph_test.py:182
↓ 1 callersFunction_create_graph_def_while_loop
------------------ px -> Mul -> | -> Add -> Mul -> | -> Relu -> Sqrt / | / / | py - c ->
tensorflow_blade/tests/util/simple_graph_test.py:123
↓ 1 callersFunction_create_graph_def_with_quirky_type_attr
px -> Add --> Mul -> Relu -> ABS -> Add -> Sqrt / \_/ / \ # noqa: W605 py -
tensorflow_blade/tests/util/simple_graph_test.py:159
↓ 1 callersMethod_create_model_args
(self, *args)
examples/PyTorch/Inference/hf_transformers/blade_adapter.py:167
↓ 1 callersFunction_deduce_from_nvcc
()
scripts/python/common_setup.py:448
↓ 1 callersFunction_deduce_from_nvcc
()
pytorch_blade/common_setup.py:448
↓ 1 callersFunction_deduplicate_onnx_graph_outputs
(graph)
pytorch_blade/torch_blade/onnx_backends/backend_conversion.py:23
↓ 1 callersFunction_deep_copy_script_module
(model)
pytorch_blade/torch_blade/exporter.py:73
↓ 1 callersFunction_delete_deepcopy_func
(mod)
pytorch_blade/torch_blade/exporter.py:78
↓ 1 callersFunction_disc_engine_conversion
(module)
pytorch_blade/torch_blade/mlir/disc_engine_conversion.py:140
↓ 1 callersFunction_erase_input_concrete_types
(graph)
pytorch_blade/torch_blade/pass_manager.py:313
↓ 1 callersFunction_fetch_dims
(tree)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/utils.py:67
↓ 1 callersFunction_fixup_for_dynamic_shape
(cfg, c_module)
pytorch_blade/torch_blade/pass_manager.py:334
↓ 1 callersMethod_gen_test_data
(self, annotation, random_seed, lower=1, upper=10)
pytorch_blade/tests/disc/testing_base.py:112
↓ 1 callersFunction_get_customize_op_black_list
(graph)
pytorch_blade/torch_blade/mlir/disc_engine_conversion.py:104
↓ 1 callersFunction_get_device
()
tensorflow_blade/setup.py:27
↓ 1 callersFunction_get_dynamic_axes
(input_list, dynamic_axes)
pytorch_blade/torch_blade/pass_manager.py:45
↓ 1 callersFunction_get_dynamic_settings
(c_module, trt_unsupported)
pytorch_blade/torch_blade/tensorrt/dynamic_shapes.py:159
↓ 1 callersMethod_get_dynamic_tuning_shapes_info
( self, subgraph_test_inputs_shapes_list: List[List[List[Shape]]], index: int,
tensorflow_blade/tf_blade/gpu/tf_to_trt.py:177
↓ 1 callersMethod_get_in_degree
(self)
pytorch_blade/torch_blade/algorithm/directed_graph.py:84
↓ 1 callersFunction_get_mlir_unsupported
(graph)
pytorch_blade/torch_blade/mlir/disc_engine_conversion.py:129
↓ 1 callersMethod_get_pattern_matched_map_list
(self)
tensorflow_blade/tf_blade/util/tf_hierarchy_pattern_match.py:204
↓ 1 callersFunction_get_shape_for_one_input
(c_module, inputs, trt_unsupported)
pytorch_blade/torch_blade/tensorrt/dynamic_shapes.py:71
↓ 1 callersFunction_get_tensor_rank
(x)
pytorch_blade/torch_blade/onnx_backends/onnx_symbolic_opset9_patches.py:21
↓ 1 callersFunction_get_version
()
tensorflow_blade/setup.py:56
↓ 1 callersMethod_has_path_dfs
(self, x: int, y: int, visited)
pytorch_blade/torch_blade/algorithm/directed_graph.py:132
↓ 1 callersMethod_hit_black_list
(self, node)
pytorch_blade/torch_blade/onnx_backends/backend_testbed.py:310
↓ 1 callersFunction_infer_type_from_observer
(graph, observer_owner, observers)
pytorch_blade/torch_blade/tools/shape_inference.py:107
↓ 1 callersMethod_insert_trt_engine_op
( self, index: int, main_graph: tf.GraphDef, engine_bytes: bytes, new_
tensorflow_blade/tf_blade/gpu/tf_to_trt.py:65
↓ 1 callersMethod_is_inplace
(self, node)
pytorch_blade/torch_blade/onnx_backends/backend_testbed.py:256
↓ 1 callersMethod_is_inplace_kinds
Check whether a view-kind node should be considered as an inplace node. let A defines a node whose kind is defined in self._view_kin
pytorch_blade/torch_blade/onnx_backends/backend_testbed.py:200
↓ 1 callersMethod_is_inplace_safe
This method is stateful, it cache safe inplace ops flags
pytorch_blade/torch_blade/onnx_backends/backend_testbed.py:260
↓ 1 callersFunction_is_list_of_int
(shapes)
pytorch_blade/torch_blade/config.py:52
↓ 1 callersFunction_is_list_of_int
(shape)
pytorch_blade/torch_blade/tensorrt/dynamic_shapes.py:46
↓ 1 callersFunction_is_normal_tensor
(tensor)
pytorch_blade/torch_blade/tools/shape_inference.py:83
↓ 1 callersFunction_is_tensor
(x)
pytorch_blade/torch_blade/onnx_backends/onnx_symbolic_opset9_patches.py:17
↓ 1 callersFunction_jit_pass_clean_script
(graph)
pytorch_blade/torch_blade/pass_manager.py:299
↓ 1 callersFunction_jit_pass_freeze_rank
(graph, is_training=False)
pytorch_blade/torch_blade/pass_manager.py:191
↓ 1 callersFunction_jit_pass_freeze_requires_grad
(graph)
pytorch_blade/torch_blade/pass_manager.py:214
↓ 1 callersFunction_jit_pass_onnx_constfold
(graph, params_dict)
pytorch_blade/torch_blade/pass_manager.py:180
↓ 1 callersFunction_jit_pass_quantization_postprocess
(c_module)
pytorch_blade/torch_blade/quantization/__init__.py:84
↓ 1 callersFunction_jit_pass_quantization_preprocess
(c_module)
pytorch_blade/torch_blade/quantization/__init__.py:72
↓ 1 callersFunction_jit_pass_remove_nograd
(graph)
pytorch_blade/torch_blade/pass_manager.py:281
↓ 1 callersFunction_jit_pass_replace_inplace_name
(graph)
pytorch_blade/torch_blade/pass_manager.py:402
↓ 1 callersFunction_make_executable
(path)
pytorch_blade/bazel_build.py:34
↓ 1 callersMethod_non_linear_param_search
r"""Non-linear parameter search. An approximation for L2 error minimization for selecting min/max. By selecting new min/max, we filter
tools/torch_quant/torch_quant/observer.py:502
↓ 1 callersFunction_operate_observer_tag
(observers)
pytorch_blade/torch_blade/tools/low_precision_analysis.py:23
↓ 1 callersMethod_opm
(self, a, b, norm)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/outer_product_mean.py:55
↓ 1 callersMethod_optimize_pattern
(self)
tensorflow_blade/tf_blade/util/tf_hierarchy_pattern_match.py:218
↓ 1 callersFunction_parent_name
Turn 'foo.bar.name' into ['foo.bar', 'name']
tools/torch_quant/torch_quant/module.py:156
↓ 1 callersMethod_patch_inputs_scalar_types
Add scalar type and dim to plain TensorType.
pytorch_blade/torch_blade/onnx_backends/backend_testbed.py:42
↓ 1 callersMethod_pattern_traversal
(self, p_node: SimpleNode, visited: Set[str] = set())
tensorflow_blade/tf_blade/util/tf_hierarchy_pattern_match.py:166
↓ 1 callersMethod_post_process_function
(self, i: int, main_graph: tf.GraphDef)
tensorflow_blade/tf_blade/gpu/tf_to_trt.py:205
↓ 1 callersMethod_prep_inputs
(self, m: torch.Tensor, z: Optional[torch.Tensor], mask: Optional[torch.Tensor], )
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/msa.py:89
↓ 1 callersMethod_prep_qkv
(self, q_x: torch.Tensor, kv_x: torch.Tensor)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/ops.py:327
↓ 1 callersFunction_process_aten_fake_quant
(c_module)
pytorch_blade/torch_blade/quantization/__init__.py:42
↓ 1 callersMethod_process_subgraph
( self, subgraph_test_inputs_shapes_list: List[List[List[Shape]]], i: int, ori
tensorflow_blade/tf_blade/gpu/tf_to_trt.py:218
↓ 1 callersMethod_process_subgraph_inputs
Add input tensors to the subgraph.
tensorflow_blade/tf_blade/util/simple_graph.py:528
↓ 1 callersMethod_process_subgraph_outputs
Add output nodes to this subgraph.
tensorflow_blade/tf_blade/util/simple_graph.py:588
↓ 1 callersFunction_prod
(nums)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/modules/ops.py:24
↓ 1 callersMethod_record_inputs_scalar_types
(self, graph)
pytorch_blade/torch_blade/onnx_backends/backend_testbed.py:51
↓ 1 callersFunction_record_shape_information
(s_module, inputs)
pytorch_blade/torch_blade/exporter.py:30
↓ 1 callersFunction_recursive_optimize
(model)
pytorch_blade/torch_blade/optimization.py:20
↓ 1 callersFunction_register_onnx_engine
(module, subgraph, state, group_name, disable_fallback=False)
pytorch_blade/torch_blade/onnx_backends/backend_conversion.py:109
↓ 1 callersMethod_remove_cycles
(self, u: int)
pytorch_blade/torch_blade/clustering/support_fusion_algorithm.py:76
← previousnext →1,701–1,800 of 6,088, ranked by callers