Searched refs:log_softmax_vulkan (Results 1 – 1 of 1) sorted by relevance
4516 const auto log_softmax_vulkan = at::log_softmax(in_vulkan, 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()