Searched refs:strategyPtr (Results 1 – 6 of 6) sorted by relevance
451 auto strategyPtr = distributed_operation_info->strategy(); in DrawParallelInfo() local452 if (strategyPtr != nullptr) { in DrawParallelInfo()456 …std::vector<ValuePtr> temp = {MakeValue(strategyPtr->GetInputStage()), MakeValue(strategyPtr->GetI… in DrawParallelInfo()
254 StrategyPtr strategyPtr; in SetStrategyToOperator() local256 strategyPtr = parallel::ExtractStrategy(attrs[STRATEGY]); in SetStrategyToOperator()258 strategyPtr = (*stra_map)[strategy_key_name]; in SetStrategyToOperator()260 if (strategyPtr != nullptr) { in SetStrategyToOperator()266 if (operator_info->SetCostUnderStrategy(strategyPtr) != SUCCESS) { in SetStrategyToOperator()275 (void)configured_stra_ops_.emplace(operator_info, strategyPtr); in SetStrategyToOperator()287 (void)configured_stra_ops_.emplace(operator_info, strategyPtr); in SetStrategyToOperator()
1318 StrategyPtr strategyPtr; in ExtractStrategy() local1343 strategyPtr = NewStrategy(stage_id, strategy); in ExtractStrategy()1346 return strategyPtr; in ExtractStrategy()1841 StrategyPtr strategyPtr; in GenerateBatchParallelStrategy() local1844 strategyPtr = NewStrategy(0, *strategy_v_ptr); in GenerateBatchParallelStrategy()1855 return strategyPtr; in GenerateBatchParallelStrategy()1912 StrategyPtr strategyPtr = nullptr; in ExtractInformation() local1932 strategyPtr = GenerateBatchParallelStrategy(operator_, prim); in ExtractInformation()1934 strategyPtr = ExtractStrategy(cnode->GetPrimalAttr(STRATEGY)); in ExtractInformation()1936 strategyPtr = ExtractStrategy(attrs[STRATEGY]); in ExtractInformation()[all …]
158 void SetLastNodeStrategy(const StrategyPtr strategyPtr);
353 StrategyPtr strategyPtr = parallel::NewStrategy(0, strategy); in TEST_F() local360 matmul_info->Init(strategyPtr); in TEST_F()508 StrategyPtr strategyPtr = parallel::NewStrategy(0, strategy); in TEST_F() local514 matmul_info->Init(strategyPtr); in TEST_F()
153 StrategyPtr strategyPtr; in TEST_F() local