Searched refs:s_data (Results 1 – 4 of 4) sorted by relevance
132 AccT* s_data = reinterpret_cast<AccT*>(s_buf); in BiasGradNHWC_SharedAtomics() local134 s_data[index] = AccT(0); in BiasGradNHWC_SharedAtomics()141 GpuAtomicAdd(s_data + bias_offset, AccT(ldg(output_backprop + index))); in BiasGradNHWC_SharedAtomics()146 GpuAtomicAdd(bias_backprop + index, T(s_data[index])); in BiasGradNHWC_SharedAtomics()157 __shared__ AccT s_data[kSDataSize]; in BiasGradNCHW_SharedAtomics() local159 s_data[index] = AccT(0); in BiasGradNCHW_SharedAtomics()181 GpuAtomicAdd(s_data + bias_offset, sum); in BiasGradNCHW_SharedAtomics()189 AccT data = s_data[thread_index]; in BiasGradNCHW_SharedAtomics()198 if (thread_index < 16) s_data[thread_index] += s_data[thread_index + 16]; in BiasGradNCHW_SharedAtomics()199 if (thread_index < 8) s_data[thread_index] += s_data[thread_index + 8]; in BiasGradNCHW_SharedAtomics()[all …]
54 const T* s_data = s.data(); in LevenshteinDistance() local79 const int replacement_cost = cmp(s_data[i - 1], t_data[j - 1]) ? 0 : 1; in LevenshteinDistance()
23 msg = ("TestS_%s_data" % suite).encode()
2090 ShapeHandle s_data = c->input(0); in UnsortedSegmentReductionShapeFn() local2101 c->MergePrefix(s_data, s_segment_ids, &s_data, &s_segment_ids)); in UnsortedSegmentReductionShapeFn()2110 c->Subshape(s_data, c->Rank(s_segment_ids), &s_data_suffix)); in UnsortedSegmentReductionShapeFn()