MCPcopy Create free account

hub / github.com/alibaba/BladeDISC / functions

Functions6,088 in github.com/alibaba/BladeDISC

↓ 1,275 callersMethodpush_back
tao_compiler/mlir/xla/ral/context/pdll_util.h:160
↓ 763 callersMethodsize
Returns the size of the ops this fusion pattern contains.
tao_compiler/mlir/disc/transforms/fusion_utils.h:233
↓ 662 callersMethodsize
tao_compiler/mlir/xla/ral/context/pdll_util.h:158
↓ 629 callersFunctionfeature_test_main
tao_compiler/mlir/disc/tests/mlir_feature_test.cc:89
↓ 519 callersMethodend
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:69
↓ 373 callersMethodgetRank
tao_compiler/mlir/xla/ral/context/common_context_impl.cc:652
↓ 372 callersMethodbegin
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:53
↓ 357 callersMethodinsert
Inserts a new pair <key, value>. Returns false if failed (e.g. duplicated key).
tao_compiler/mlir/xla/ral/context/pdll_util.h:137
↓ 339 callersMethodend
tao/tao_bridge/tf/resource_operation_safety_analysis.cc:208
↓ 338 callersMethodassertEqual
(self, x, y, *args, **kwargs)
pytorch_blade/torch_blade/testing/common_utils.py:97
↓ 304 callersMethodgetContext
tao_compiler/mlir/xla/ral/ral_context.cc:238
↓ 287 callersMethodappend
(self, name, secs)
pytorch_blade/common_setup.py:47
↓ 287 callersMethodtype
tao/tao_bridge/tf/tf2xla_util.h:66
↓ 261 callersMethodname
tao/tao_bridge/passes/functionalize_cond.cc:805
↓ 229 callersMethodgetOperation
tao_compiler/mlir/disc/transforms/shape_utils.h:237
↓ 222 callersMethodbegin
tao/tao_bridge/tf/resource_operation_safety_analysis.cc:204
↓ 218 callersMethodget
tao_compiler/mlir/xla/ral/context/pdll_util.h:172
↓ 209 callersMethodempty
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:94
↓ 208 callersMethodnode
Get node with given index.
tensorflow_blade/tf_blade/util/simple_graph.py:126
↓ 207 callersMethodkind
tao/tao_bridge/tf/deadness_analysis.cc:201
↓ 181 callersMethodgetShape
tao_compiler/mlir/disc/transforms/shape_utils.cc:640
↓ 171 callersMethodgetValue
tao_compiler/mlir/disc/transforms/shape_utils.h:83
↓ 163 callersMethodfind
tao_compiler/mlir/xla/ral/ral_context.cc:144
↓ 160 callersMethodstr
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:147
↓ 158 callersMethodclear
tao_compiler/mlir/xla/ral/ral_context.cc:63
↓ 158 callersMethodempty
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.h:71
↓ 141 callersMethodgetResults
tao_compiler/mlir/disc/transforms/disc_comp_intens_fusion_to_cuda_source.cc:313
↓ 140 callersMethodget
tao_compiler/mlir/disc/transforms/disc_convert_fake_quant_op.cc:47
↓ 134 callersMethodsignalError
tao_compiler/mlir/xla/ral/ral_context.cc:174
↓ 129 callersMethodok
tao_compiler/mlir/xla/ral/ral_context.cc:68
↓ 121 callersMethoderase
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:122
↓ 116 callersMethodid
tao/tao_bridge/tf/device_util.h:55
↓ 113 callersMethoddata
tao_compiler/mlir/xla/ral/context/base/base_context.h:52
↓ 108 callersMethodfind
Returns true if found, otherwise return false. When found, `out_filename` will be filled with the filename of the compiled result proto corresponding
tao/tao_bridge/kernels/tao_compilation_cache.cc:375
↓ 107 callersMethodnodes
Get all nodes in graph.
tensorflow_blade/tf_blade/util/simple_graph.py:114
↓ 104 callersMethodtensor
Constructs an tensor with no buffer and zero memory description
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:443
↓ 102 callersMethodformat
(self, record)
pytorch_blade/common_setup.py:113
↓ 97 callersMethodget
(self, key, default_value=None)
pytorch_blade/benchmark/torch-tensorrt/perf_run.py:59
↓ 76 callersMethodregister_buffer
(self, full_path: str, value: torch.Tensor)
tools/torch_quant/torch_quant/graph.py:176
↓ 75 callersMethodfind
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:73
↓ 73 callersMethodformat
(self, record)
tensorflow_blade/common_internal.py:45
↓ 64 callersMethodgetDimSize
tao_compiler/mlir/disc/transforms/shape_utils.h:32
↓ 63 callersFunctionReadBoolFromEnvVar
pytorch_blade/pytorch_blade/common_utils/utils.cpp:87
↓ 62 callersMethodexists
(self, path)
scripts/python/bazel_snapshot_client.py:54
↓ 61 callersMethodgetValue
Returns the value of the attr
tao_compiler/mlir/xla/ral/context/pdll_util.h:66
↓ 61 callersMethodrun
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/graph_fuser.cpp:1163
↓ 58 callersMethodget_current_context_or_new
Return the deepest context on the stack or create a new Config.
pytorch_blade/torch_blade/config.py:518
↓ 58 callersMethodget_desc
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:427
↓ 58 callersMethodin_edges
(self, node: int)
pytorch_blade/torch_blade/algorithm/directed_graph.py:39
↓ 55 callersMethodgetName
tao_compiler/mlir/disc/transforms/fusion_utils.h:502
↓ 55 callersFunctionisEmptyMemref
tao_compiler/mlir/xla/ral/context/context_util.h:84
↓ 53 callersMethodFind
tao_compiler/mlir/xla/ral/ral_helper.cc:53
↓ 52 callersMethodmap
tao_compiler/mlir/disc/transforms/lhlo_elemental_utils.h:192
↓ 49 callersMethodout_edges
(self, node: int)
pytorch_blade/torch_blade/algorithm/directed_graph.py:42
↓ 48 callersMethod_test_cvt_to_disc
( self, nn_module, test_data, annotations=[], rtol=1e-6, atol=1e-3, n_engines=1 )
pytorch_blade/tests/disc/testing_base.py:93
↓ 48 callersMethodclone
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:235
↓ 47 callersMethodresize
Recreate a param with completely different content from old one but reuse the param shell. Notice that after resize, its format is undefined legacy AP
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:716
↓ 47 callersMethodsize
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:38
↓ 46 callersMethodfind
(self, x: int)
pytorch_blade/torch_blade/algorithm/union_set.py:26
↓ 42 callersMethodshape
(x)
pytorch_blade/tests/test_tools.py:74
↓ 41 callersMethodformat
(self, record)
scripts/python/common_setup.py:113
↓ 41 callersMethodop
(x)
pytorch_blade/tests/tensorrt/test_support_info.py:147
↓ 40 callersMethodtrace
(self, root, **kwargs)
tools/torch_quant/bert_ptq_demo.py:38
↓ 39 callersMethodAddEdge
tao/tao_bridge/tf/graphcycles_test.cc:402
↓ 39 callersMethodgetNumElements
Returns the total number of elements of the denseElementsAttr.
tao_compiler/mlir/xla/ral/context/pdll_util.h:106
↓ 37 callersFunctionGetTaoBridgeOptions
tao/tao_bridge/common.cc:131
↓ 37 callersMethod_test_disc
(self, nn_module, annotation, test_data=None, rtol=1e-6, atol=1e-3, random_seed=10)
pytorch_blade/tests/disc/testing_base.py:123
↓ 37 callersMethodcreate
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/lru_cache.hpp:148
↓ 37 callersMethodinit
Function that refill tensor with new description. Specifiy extra buffer.
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:488
↓ 36 callersFunctionOkStatus
tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc:42
↓ 36 callersFunctionset_attr_dtype
(node: tf.NodeDef, key: str, value: tf.DType)
tensorflow_blade/tf_blade/util/tf_graph_transform_util.py:103
↓ 35 callersFunction_action_env
(key, value, cmd="build")
tensorflow_blade/build.py:128
↓ 34 callersMethodToString
tao/tao_bridge/tf/deadness_analysis.cc:188
↓ 34 callersFunctiongetShapeValues
tao_compiler/mlir/disc/disc_util.cc:436
↓ 34 callersMethodget_dims
Returns dimension vector
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:599
↓ 34 callersMethodrelu
(x: torch.Tensor)
pytorch_blade/tests/test_tools.py:49
↓ 33 callersMethodload
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:84
↓ 32 callersFunction_action_env
(key, value, cmd="build")
scripts/python/tao_build.py:137
↓ 32 callersMethodsave
tao_compiler/mlir/disc/transforms/disc_shape_optimization_utils.cc:417
↓ 30 callersFunctionSize
tao_compiler/mlir/xla/ral/context/context_util.h:95
↓ 30 callersFunctionbitonicExchange256
tao_compiler/mlir/xla/ral/context/custom_library/bitonic_sort.cu.h:98
↓ 30 callersFunctionbitonicExchange256Pair
tao_compiler/mlir/xla/ral/context/custom_library/bitonic_sort.cu.h:84
↓ 30 callersMethodget_data_type
Returns descriptor data type
tao_compiler/mlir/xla/ral/context/mkldnn/ideep/ideep/tensor.hpp:609
↓ 29 callersFunctionbuildMatchOp
tao_compiler/mlir/disc/transforms/disc_transform_schedule.cc:96
↓ 29 callersFunctionhash_combine
tao_compiler/mlir/xla/ral/context/quantized_gpu_kernel_impl.cc:103
↓ 28 callersFunctionGetI64ElementsAttr
tao_compiler/mlir/disc/disc_util.h:51
↓ 28 callersFunctionReadStringFromEnvVar
pytorch_blade/pytorch_blade/common_utils/utils.cpp:123
↓ 28 callersMethodgetImpl
tao_compiler/mlir/xla/ral/context/tensorflow/tf_context_impl.h:62
↓ 28 callersFunctionprint_memref
tao_compiler/mlir/xla/ral/context/context_util.h:24
↓ 28 callersMethodtranspose
(x)
pytorch_blade/tests/disc/ops/test_permutation.py:33
↓ 27 callersFunctionbuildRunCanonicalizer
tao_compiler/mlir/disc/transforms/disc_transform_schedule.cc:148
↓ 27 callersFunctioncreateLoadOrUseCachedValue
tao_compiler/mlir/disc/transforms/lhlo_elemental_utils.cc:98
↓ 27 callersFunctionget_source_root_dir
()
scripts/python/common_setup.py:85
↓ 27 callersMethodrun
(self)
tensorflow_blade/setup.py:100
↓ 25 callersFunctionexecute
Execute a shell command, exception raised on failure.
pytorch_blade/common_setup.py:146
↓ 24 callersFunction_add_op_common
( graph_def: tf.GraphDef, op_type: OpType, input_name_list: List[str], name: str )
tensorflow_blade/tf_blade/util/tf_graph_transform_util.py:169
↓ 24 callersFunctioncreateMaySpecificLoad
tao_compiler/mlir/disc/transforms/lhlo_elemental_utils.cc:85
↓ 24 callersFunctionirange
pytorch_blade/pytorch_blade/compiler/jit/torch/shape_analysis.cpp:51
↓ 24 callersFunctionscatter
(input: Tensor, dim: int = -1)
examples/PyTorch/Inference/CUDA/AlphaFold/FoldAcc/foldacc/model/distributed/comm.py:61
↓ 23 callersMethodLookupCondId
tao/tao_bridge/passes/functionalize_cond.cc:188
next →1–100 of 6,088, ranked by callers