Searched refs:log_softmax_cpu (Results 1 – 1 of 1) sorted by relevance
4515 const auto log_softmax_cpu = at::log_softmax(in_cpu, 0); in TEST_F() local4518 …const auto check = checkRtol(log_softmax_cpu - log_softmax_vulkan.cpu(), -smallest_representable_l… in TEST_F()4520 showRtol(log_softmax_cpu, log_softmax_vulkan.cpu()); in TEST_F()