Searched refs:values_slice (Results 1 – 4 of 4) sorted by relevance
56 auto values_slice = values_acc[b]; in sampled_addmm_sparse_csr_kernel_impl() local66 scalar_t val = values_slice[e]; in sampled_addmm_sparse_csr_kernel_impl()74 values_slice[e] = val; in sampled_addmm_sparse_csr_kernel_impl()
190 V *values_slice = &values.data[valueStartOffset]; in C10_LAUNCH_BOUNDS_1() local193 StridedRandomAccessor<V, IndexType> values_iter(values_slice, valueSliceStride); in C10_LAUNCH_BOUNDS_1()274 V *values_slice = &values.data[valueStartOffset];277 StridedRandomAccessor<V, IndexType> values_iter(values_slice, valueSliceStride);
85 for values_slice, index in zip(a.values, a.indices):88 arr[index] += values_slice
2400 TF_ASSIGN_OR_RETURN(const BufferAllocation::Slice values_slice, in HandleTopK()2407 EmitBufferPointer(values_slice, hlo->operand(0)->shape()); in HandleTopK()