Home
last modified time | relevance | path

Searched refs:distance_functions (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dmodules.cpp2208 distance_functions = {pairwise_distance, cosine_distance}; in TEST_F() local
2215 for (auto& function : distance_functions) { in TEST_F()
/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py12256 distance_functions = (pairwise_distance, cosine_distance,
12264 in itertools.product(distance_functions, reductions, margins, swaps):