Searched refs:StackInfo (Results 1 – 5 of 5) sorted by relevance
31 Status StackInfo::GetAttrs() { in GetAttrs()60 Status StackInfo::CheckStrategy(const StrategyPtr &strategy) { in CheckStrategy()86 Status StackInfo::InferDevMatrixShape() { in InferDevMatrixShape()98 Status StackInfo::InferTensorMap() { in InferTensorMap()122 void StackInfo::ReComputeBatchSplitFlagList() { in ReComputeBatchSplitFlagList()128 Status StackInfo::SetCostUnderStrategy(const StrategyPtr &strategy) { return SetCostUnderStrategyBa… in SetCostUnderStrategy()130 std::vector<StrategyPtr> StackInfo::GenerateOpStrategies(int64_t stage_id) { in GenerateOpStrategies()164 Status StackInfo::Init(const StrategyPtr &strategy) { in Init()173 Status StackInfo::InitForCostModel(const StrategyPtr &strategy) { in InitForCostModel()
32 class StackInfo : public OperatorInfo {34 …StackInfo(const std::string &operator_name, const Shapes &inputs_shape, const Shapes &outputs_shap… in StackInfo() function37 ~StackInfo() override = default;56 using PackInfoPtr = std::shared_ptr<StackInfo>;
308 struct StackInfo { struct in Graph313 SkTArray<StackInfo> fStack; // only used in addEdges()
191 REGISTER(StackInfo);