Searched defs:parallel_shape (Results 1 – 3 of 3) sorted by relevance
515 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SetClonedTensorShapeForOptimizer() local607 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in HandleAdaFactorOpt() local
1693 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SetParallelShape() local2382 …std::shared_ptr<abstract::BaseShape> parallel_shape = std::make_shared<abstract::Shape>(slice_shap… in SplitSens() local
683 BaseShapePtr parallel_shape = std::make_shared<abstract::Shape>(slice_shape); in InsertReceive() local