Searched defs:split_num (Results 1 – 8 of 8) sorted by relevance
31 int64_t split_num = 0; in Parse() local
89 int64_t split_num = split_info->out_num; in CalSplitOutputShape() local134 int64_t split_num = split_info->out_num; in CalSplitInShape() local248 size_t split_num = split_info->size_splits.size(); in UpdateSplitInfo() local
143 int64_t split_num = 1; in ReplaceNodeInputOrAttrs() local
59 int64_t split_num = static_cast<int64_t>(splits.size()); in CreateSplitConstantTensors() local332 … std::vector<AnfNodePtr> *split_outputs, size_t split_num, in CreateOutputsOfSplit()
145 … std::vector<AnfNodePtr> *split_outputs, size_t split_dim, size_t split_num, in CreateOutputsOfSplit()
435 int64_t split_num; in GenerateOptShardSliceShape() local
240 int32_t split_num = 0; in ParseLine() local
1988 def __infer__(self, dy, split_num): argument