Searched refs:test_elements_flat (Results 1 – 3 of 3) sorted by relevance
319 Tensor test_elements_flat = test_elements.to(common_type).view(-1); in isin_default_kernel_cpu() local320 auto test_elements_stride = test_elements_flat.stride(0); in isin_default_kernel_cpu()330 const auto* test_element_data = test_elements_flat.const_data_ptr<scalar_t>(); in isin_default_kernel_cpu()331 for (const auto j : c10::irange(test_elements_flat.numel())) { in isin_default_kernel_cpu()
459 Tensor elements_flat, test_elements_flat, unique_order; in isin_sorting() local462 test_elements_flat = test_elements.ravel(); in isin_sorting()466 std::tie(test_elements_flat, std::ignore) = at::_unique(test_elements, /*sorted=*/ false); in isin_sorting()472 Tensor all_elements = at::cat({std::move(elements_flat), std::move(test_elements_flat)}); in isin_sorting()
5046 test_elements_flat = test_elements.flatten()5051 all_elements = torch.cat([elements_flat, test_elements_flat])5065 sorted_test_elements, _ = torch.sort(test_elements_flat)