Searched refs:curr_segment (Results 1 – 3 of 3) sorted by relevance
96 IntType curr_segment = segment; in concat_variable_kernel() local99 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) { in concat_variable_kernel()101 ++curr_segment; in concat_variable_kernel()106 const T* input_ptr = input_ptrs[curr_segment]; in concat_variable_kernel()
151 IntType curr_segment = segment; in split_v_kernel() local154 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) { in split_v_kernel()156 ++curr_segment; in split_v_kernel()161 T* output_ptr = output_ptrs[curr_segment]; in split_v_kernel()
843 auto& curr_segment = initial_segments.at(t); in ConvertAfterShapes() local847 GetEngineInfo(&graph, *params.graph_properties, curr_segment.first, in ConvertAfterShapes()873 total_num_nodes_in_segments += curr_segment.first.size(); in ConvertAfterShapes()875 converted_segments.push_back(std::move(curr_segment)); in ConvertAfterShapes()