Searched refs:current_indices_offset (Results 1 – 1 of 1) sorted by relevance
182 size_t current_indices_offset = 0; in SplitAndCalculateSegmentBucketSize() local190 segments[i]->value_ = input_grad->value_ + current_indices_offset * param.value_stride_; in SplitAndCalculateSegmentBucketSize()191 segments[i]->indices_ = input_grad->indices_ + current_indices_offset; in SplitAndCalculateSegmentBucketSize()198 current_indices_offset += indices_size; in SplitAndCalculateSegmentBucketSize()248 size_t current_indices_offset = 0; in GatherSegmentIndicesToOutputBucket() local251 … buckets[i]->value_ = param.output_grad_->value_ + current_indices_offset * param.value_stride_; in GatherSegmentIndicesToOutputBucket()252 buckets[i]->indices_ = param.output_grad_->indices_ + current_indices_offset; in GatherSegmentIndicesToOutputBucket()253 buckets[i]->global_indices_ = param.workspace_grad_->indices_ + current_indices_offset; in GatherSegmentIndicesToOutputBucket()255 current_indices_offset += bucket_data_size[i]; in GatherSegmentIndicesToOutputBucket()273 current_indices_offset = 0; in GatherSegmentIndicesToOutputBucket()[all …]