Searched defs:split_count (Results 1 – 9 of 9) sorted by relevance
60 int64_t split_count = AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateSplitNode() local97 int64_t split_count = AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateAllToAllvNode() local130 int64_t split_count = AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateConcatNode() local
82 int64_t split_count = args[0]; in TEST_F() local125 int64_t split_count = 2; in TEST_F() local130 int64_t split_count = 2; in TEST_F() local
90 int64_t split_count = args[TRANSFER_PERMUTE_SPLIT_COUNT_INDEX]; in StridedSliceOP() local178 Status ConstructOperator::SplitOP(int64_t split_count) { in SplitOP()201 int64_t split_count = args[TRANSFER_PERMUTE_SPLIT_COUNT_INDEX]; in AlltoAllOP() local
263 int64_t split_count = args[TRANSFER_CONCAT_SPLIT_COUNT_INDEX]; in TransferConcatByAxis() local
61 std::size_t split_count = L / 2; in generate_elements() local
41 int split_count = 0; in CheckStrategy() local
131 int split_count = 0; in CheckStrategy() local
681 def __init__(self, split_count, split_dim, concat_dim, group=GlobalComm.WORLD_COMM_GROUP): argument
1469 BOOST_INTRUSIVE_FORCEINLINE SizeType split_count() const in split_count() function