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()
147 IntType curr_segment = segment; in split_v_kernel() local150 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) { in split_v_kernel()152 ++curr_segment; in split_v_kernel()157 T* output_ptr = output_ptrs[curr_segment]; in split_v_kernel()
767 auto& curr_segment = initial_segments.at(t); in ConvertAfterShapes() local770 Status status = GetEngineInfo(&graph, static_graph_properties, curr_segment, in ConvertAfterShapes()797 converted_segments.push_back(std::move(curr_segment)); in ConvertAfterShapes()