MCPcopy Create free account

hub / github.com/alibaba/BladeDISC / functions

Functions6,088 in github.com/alibaba/BladeDISC

⨍ Functions 6,088◇ Types & classes 1,279↳ Endpoints 6
↓ 1 callersFunctionGetTestingTensor
tao_compiler/mlir/disc/tools/disc-replay/replay_test.cc:22
↓ 1 callersFunctionGetTransposeOutputType
tao_compiler/mlir/disc/transforms/quantized_dot_rewriter.cc:53
↓ 1 callersFunctionGetWhitelistTable
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:2709
↓ 1 callersFunctionGetXlaClusterInfo
tao/tao_bridge/passes/tao_build_tao_op_pass.cc:55
↓ 1 callersFunctionGetXlaGlobalJitLevel
tao/tao_bridge/tf/xla_cluster_util.cc:231
↓ 1 callersFunctionHandleCpuFusionOp
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:5400
↓ 1 callersFunctionHandleCpuFusionOp
tao_compiler/mlir/disc/transforms/disc_stitch_fusion.cc:42
↓ 1 callersFunctionHandleGpuFusionOp
tao_compiler/mlir/disc/transforms/lhlo_legalize_roots_to_loops.cc:3889
↓ 1 callersFunctionHasCandidateBroadcastOp
tao_compiler/mlir/disc/transforms/disc_specialize_fusion_with_speculation.cc:76
↓ 1 callersFunctionHasForwardedRefInput
tao/tao_bridge/tf/xla_cluster_util.cc:95
↓ 1 callersFunctionHasFunctionalControlFlowOps
tao/tao_bridge/tf/compilability_check_util.cc:98
↓ 1 callersMethodHasGradientOp
graph properties query
tao/tao_bridge/passes/tao_feature_detector.h:31
↓ 1 callersFunctionHasResourceInput
tao/tao_bridge/tf/compilability_check_util.cc:69
↓ 1 callersFunctionHumanReadableNumFlops
tao/tao_bridge/tf/util.cc:175
↓ 1 callersFunctionIgnoreResourceOpForSafetyAnalysis
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:606
↓ 1 callersFunctionIncomingEdgeAsOutput
tao/tao_bridge/passes/tao_defuse_pass.cc:46
↓ 1 callersMethodInit
tao_compiler/mlir/xla/ral/context/base/cuda/cuda_stream.cc:20
↓ 1 callersFunctionInitGPU
tao_compiler/mlir/disc/disc_compiler_main.cc:132
↓ 1 callersFunctionInitLtcModuleBindings
pytorch_blade/pytorch_blade/ltc/init_python_bindings.cpp:19
↓ 1 callersFunctionInitModule
tao_compiler/decoupling/mlir_compiler_impl_gpu.cc:74
↓ 1 callersFunctionInitModule
tao_compiler/decoupling/mlir_compiler_impl_cpu.cc:34
↓ 1 callersFunctionInitModule
tao_compiler/decoupling/mlir_compiler_impl_rocm.cc:63
↓ 1 callersFunctionInitRngWithRandomSeed
tao_compiler/mlir/xla/ral/context/random_impl.cc:37
↓ 1 callersMethodInitShapeInfo
tao/tao_bridge/kernels/tao_compilation_info_collector.cc:494
↓ 1 callersMethodInlineCallNodes
tao/tao_bridge/tf/lower_if_op.cc:240
↓ 1 callersFunctionInsertAlloc
tao_compiler/mlir/disc/transforms/disc_hlo_legalize_to_lhlo.cc:89
↓ 1 callersFunctionInsertDynamicAlloc
tao_compiler/mlir/disc/transforms/disc_hlo_legalize_to_lhlo.cc:61
↓ 1 callersFunctionInsertSyncOnStream
Insert a sync on stream call.
tao_compiler/mlir/disc/transforms/disc_lower_to_library_call.cc:101
↓ 1 callersFunctionInsertTranspose
tao_compiler/mlir/disc/transforms/quantized_dot_rewriter.cc:66
↓ 1 callersFunctionIsActivation
tao/tao_bridge/passes/tao_defuse_pass.cc:39
↓ 1 callersMethodIsCompilableCall
Tests whether 'call_def' is a call to a completely compilable function. Every operator in the function must be compilable for a function to be compila
tao/tao_bridge/tf/compilability_check_util.cc:293
↓ 1 callersMethodIsCompilableNode
tao/tao_bridge/tf/compilability_check_util.cc:368
↓ 1 callersMethodIsCpu
tao/tao_bridge/tf/device_util.h:117
↓ 1 callersFunctionIsEdgeSafe
Returns true if a control or data dependence from a TensorFlow operation of resource op kind `from` to a TensorFlow operation of resource op kind `to`
tao/tao_bridge/tf/resource_operation_safety_analysis.cc:137
↓ 1 callersFunctionIsEmptyTensor
pytorch_blade/pytorch_blade/compiler/mlir/runtime/ral_context.cpp:175
↓ 1 callersFunctionIsFakeQuantArgumentConstant
almost same with IsSmallHostConstant but skip the check of op's size for fakequant's scale and zero_point
tao/tao_bridge/passes/tao_clone_constants_for_better_clustering.cc:98
↓ 1 callersMethodIsGpu
tao/tao_bridge/tf/device_util.h:116
↓ 1 callersFunctionIsInPlaceOp
tao/tao_bridge/passes/tao_clone_constants_for_better_clustering.cc:116
↓ 1 callersMethodIsInRange
pytorch_blade/pytorch_blade/compiler/tensorrt/tensorrt_engine_context.cpp:230
↓ 1 callersFunctionIsMlirMhloSupported
pytorch_blade/pytorch_blade/compiler/mlir/converters/mhlo_conversion.cpp:162
↓ 1 callersFunctionIsMustCompileDevice
tao/tao_bridge/passes/tao_partially_decluster_pass.cc:241
↓ 1 callersFunctionIsSingleGpuGraph
tao/tao_bridge/tf/xla_cluster_util.cc:269
↓ 1 callersFunctionIsSmallCpuAlloc
tao_compiler/mlir/disc/disc_util.cc:49
↓ 1 callersMethodIsSupportedModel
tensorflow_blade/src/tensorrt/bridge/tensorrt_onnx_parser.cpp:175
↓ 1 callersMethodIsSupportedModel
pytorch_blade/pytorch_blade/compiler/tensorrt/bridge/tensorrt_onnx_parser.cpp:183
↓ 1 callersFunctionIsSupportedResourceForXLA
tao/tao_bridge/passes/tao_mark_for_compilation_pass.cc:2076
↓ 1 callersFunctionIsTorchMlirSupported
pytorch_blade/pytorch_blade/compiler/mlir/converters/torch_mlir_op_filter.cpp:26
↓ 1 callersFunctionIsXlaCompiledKernel
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:2803
↓ 1 callersMethodKill
tao/tao_bridge/tf/subprocess.cc:288
↓ 1 callersFunctionLaunchTopKKernel
tao_compiler/mlir/xla/ral/context/custom_library/tf_topk.cu.h:360
↓ 1 callersMethodLoad
tao_compiler/mlir/disc/tools/disc-replay/record.cc:51
↓ 1 callersMethodLock
tao/tao_bridge/kernels/tao_compilation_cache.cc:110
↓ 1 callersFunctionLogError
Log the error at the given severity, optionally with a stack trace. If log_severity is NUM_SEVERITIES, nothing is logged.
tao/tao_bridge/errors.cc:26
↓ 1 callersFunctionLogLines
tao/tao_bridge/tf/util.cc:183
↓ 1 callersFunctionLoopParallelAssigner
tao_compiler/mlir/xla/ral/context/common_context_impl.cc:782
↓ 1 callersMethodLower
pytorch_blade/pytorch_blade/ltc/include/torch/csrc/lazy/ts_backend/ops/to_copy.h:91
↓ 1 callersFunctionLowerHLOToLLVM
tao_compiler/mlir/disc/disc_compiler.cc:210
↓ 1 callersFunctionLowerLLVMToBinary
tao_compiler/mlir/disc/disc_compiler.cc:802
↓ 1 callersMethodMakeAndRecurrencePredicate
tao/tao_bridge/tf/deadness_analysis.cc:399
↓ 1 callersFunctionMakeCallNodeFromAttribute
tao/tao_bridge/tf/compilability_check_util.cc:80
↓ 1 callersMethodMakeComputationDataFromScalar
pytorch_blade/pytorch_blade/ltc/disc_backend/backend_impl.cpp:137
↓ 1 callersMethodMakeComputationDataFromTensor
pytorch_blade/pytorch_blade/ltc/disc_backend/backend_impl.cpp:116
↓ 1 callersMethodMakeNodeImage
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:755
↓ 1 callersFunctionMakeStatus
tao/tao_bridge/errors.cc:20
↓ 1 callersMethodMakeTrue
tao/tao_bridge/tf/deadness_analysis.cc:487
↓ 1 callersFunctionMakeUniqueFilename
tao/tao_bridge/tf/dump_graph.cc:38
↓ 1 callersFunctionMarkGuaranteedConstants
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:107
↓ 1 callersMethodMatch
tao_compiler/mlir/xla/ral/context/stream_executor_based_impl.cc:977
↓ 1 callersFunctionMaybeCastTo
Returns a new scalar integer value having type `type`. Here `type` must be an integer or index type.
tao_compiler/mlir/disc/IR/qconv_custom_call_op.cc:35
↓ 1 callersFunctionMaybeDefunctionalize
tao/tao_bridge/passes/functionalize_control_flow.cc:67
↓ 1 callersFunctionMaybePickDeviceForXla
tao/tao_bridge/tf/device_util.cc:240
↓ 1 callersFunctionMhloConversaion
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/register_disc_class.cpp:51
↓ 1 callersFunctionMinLogLevelFromEnv
tao_compiler/mlir/xla/ral/ral_logging.cc:119
↓ 1 callersFunctionMinVLogLevelFromEnv
tao_compiler/mlir/xla/ral/ral_logging.cc:124
↓ 1 callersFunctionMoveOutgoingEdges
tao/tao_bridge/passes/tao_defuse_pass.cc:60
↓ 1 callersFunctionMoveOutgoingEdges
tao/tao_bridge/passes/tao_build_tao_op_pass.cc:132
↓ 1 callersFunctionMustCompileNode
tao/tao_bridge/passes/tao_partially_decluster_pass.cc:251
↓ 1 callersFunctionNameAndAttrsFromFunctionCall
tao/tao_bridge/tf/compilability_check_util.cc:277
↓ 1 callersFunctionNoLoaderUser
tao_compiler/mlir/disc/transforms/fusion_utils.cc:72
↓ 1 callersMethodNotifyAndWaitToStop
tao/tao_bridge/kernels/tao_compilation_cache.cc:240
↓ 1 callersMethodNotifyCancellation
tao/tao_bridge/kernels/tao_compilation_cache.cc:117
↓ 1 callersFunctionNvDataType2String
TODO: Unify error handling methods
tensorflow_blade/src/tensorrt/bridge/tensorrt_common.cpp:43
↓ 1 callersFunctionNvDataType2String
pytorch_blade/pytorch_blade/compiler/tensorrt/bridge/tensorrt_common.cpp:42
↓ 1 callersFunctionNvDataType2TorchDataType
pytorch_blade/pytorch_blade/compiler/tensorrt/tensorrt_engine_context.cpp:26
↓ 1 callersFunctionNvInferType2TorchType
pytorch_blade/pytorch_blade/compiler/tensorrt/tensorrt_engine_context.cpp:44
↓ 1 callersMethodOutsideCompilationAncestorMap
tao/tao_bridge/passes/tao_encapsulate_subgraphs_pass.cc:903
↓ 1 callersFunctionPackRandomUniformBackendConfig
tao_compiler/mlir/disc/transforms/lower_tf.cc:61
↓ 1 callersFunctionParseInputDescriptor
tao_compiler/mlir/disc/tests/mlir_test.cc:120
↓ 1 callersFunctionParseOutputDescriptor
tao_compiler/mlir/disc/tests/mlir_test.cc:144
↓ 1 callersFunctionPartiallyDeclusterNode
tao/tao_bridge/passes/tao_partially_decluster_pass.cc:121
↓ 1 callersFunctionPeepholeOptimizeShapeExpressions
pytorch_blade/pytorch_blade/ltc/disc_compiler/passes/graph_fuser.cpp:1225
↓ 1 callersFunctionPlatformInfoFromContext
tao/tao_bridge/kernels/platform_info.cc:17
↓ 1 callersMethodPredicateMapAsString
tao/tao_bridge/tf/deadness_analysis.cc:1597
↓ 1 callersMethodPrint
tao/tao_bridge/tf/deadness_analysis.cc:1564
↓ 1 callersFunctionPrintGCTransitiveClosure
tao/tao_bridge/tf/graphcycles_test.cc:120
↓ 1 callersFunctionPrintNodes
tao/tao_bridge/tf/graphcycles_test.cc:73
↓ 1 callersFunctionPrintTransitiveClosure
tao/tao_bridge/tf/graphcycles_test.cc:104
↓ 1 callersFunctionPropagateCatShape
pytorch_blade/pytorch_blade/compiler/jit/torch/shape_analysis.cpp:521
↓ 1 callersMethodPropagateUpdatedState
tao/tao_bridge/passes/functionalize_cond.cc:842
↓ 1 callersFunctionRandomEdge
Returns the index of a randomly chosen edge in *edges. Requires *edges be non-empty.
tao/tao_bridge/tf/graphcycles_test.cc:195
← previousnext →1,501–1,600 of 6,088, ranked by callers