Searched refs:s_data (Results 1 – 4 of 4) sorted by relevance
129 AccT* s_data = reinterpret_cast<AccT*>(s_buf); in BiasGradNHWC_SharedAtomics() local131 s_data[index] = AccT(0); in BiasGradNHWC_SharedAtomics()138 CudaAtomicAdd(s_data + bias_offset, AccT(ldg(output_backprop + index))); in BiasGradNHWC_SharedAtomics()143 CudaAtomicAdd(bias_backprop + index, T(s_data[index])); in BiasGradNHWC_SharedAtomics()155 __shared__ AccT s_data[kSDataSize]; in BiasGradNCHW_SharedAtomics() local157 s_data[index] = AccT(0); in BiasGradNCHW_SharedAtomics()179 CudaAtomicAdd(s_data + bias_offset, sum); in BiasGradNCHW_SharedAtomics()186 AccT data = s_data[thread_index]; in BiasGradNCHW_SharedAtomics()
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()
1103 ShapeHandle s_data = c->input(0); in UnsortedSegmentReductionShapeFn() local1114 c->MergePrefix(s_data, s_segment_ids, &s_data, &s_segment_ids)); in UnsortedSegmentReductionShapeFn()1123 c->Subshape(s_data, c->Rank(s_segment_ids), &s_data_suffix)); in UnsortedSegmentReductionShapeFn()