Searched refs:src_flat (Results 1 – 2 of 2) sorted by relevance
174 auto src_flat = src.flat<tstring>(); in FillStringTensor() local176 dst_flat(i) = src_flat(i); in FillStringTensor()
1771 Tensor src_flat = at::as_strided( in _scatter_via_index_put() local1781 mut_out_flat.index_put_(indices, src_flat, accumulate); in _scatter_via_index_put()