Searched refs:rank_size_t (Results 1 – 2 of 2) sorted by relevance
28 size_t rank_size_t = LongToSize(rank_size); in InsertSplitForInput() local37 for (size_t j = 0; j < rank_size_t; j++) { in InsertSplitForInput()39 output_node_shape[0] /= rank_size_t; in InsertSplitForInput()65 size_t rank_size_t = LongToSize(rank_size); in RearrangeInputsForReduceScatter() local66 for (size_t i = 0; i < rank_size_t; i++) { in RearrangeInputsForReduceScatter()67 for (size_t j = 0, idx = i; j < inputs_size; j++, idx += rank_size_t) { in RearrangeInputsForReduceScatter()
54 size_t rank_size_t = LongToSize(rank_size); in GenerateKernelBuildInfo() local55 if (rank_size_t == 0) { in GenerateKernelBuildInfo()64 for (size_t rank_index = 0; rank_index < rank_size_t; ++rank_index) { in GenerateKernelBuildInfo()71 shape[0] /= rank_size_t; in GenerateKernelBuildInfo()341 size_t rank_size_t = LongToSize(rank_size); in CreateFusedCommunicationOp() local342 if (rank_size_t == 0) { in CreateFusedCommunicationOp()345 size_t output_num = node_num * rank_size_t; in CreateFusedCommunicationOp()349 for (size_t i = 0; i < rank_size_t; ++i) { in CreateFusedCommunicationOp()355 shape[0] /= rank_size_t; in CreateFusedCommunicationOp()