Searched defs:neighbors_data (Results 1 – 1 of 1) sorted by relevance
158 auto neighbors_data = xla::Gather(data, gather_indices, gather_dim_numbers, in Gather2by2Neighbors() local391 auto neighbors_data = in CalculateGradWarp() local522 auto neighbors_data = Gather2by2Neighbors( in Compile() local