Searched refs:computeFlops (Results 1 – 5 of 5) sorted by relevance
2801 flops = torch::profiler::impl::computeFlops( in TEST()2818 flops = torch::profiler::impl::computeFlops( in TEST()2827 flops = torch::profiler::impl::computeFlops( in TEST()2836 flops = torch::profiler::impl::computeFlops( in TEST()2844 flops = torch::profiler::impl::computeFlops( in TEST()2855 torch::profiler::impl::computeFlops(std::string("aten::mm"), extra_args); in TEST()2859 flops = torch::profiler::impl::computeFlops( in TEST()2870 torch::profiler::impl::computeFlops(std::string("aten::bmm"), extra_args); in TEST()2874 flops = torch::profiler::impl::computeFlops( in TEST()2881 torch::profiler::impl::computeFlops(std::string("aten::mm"), extra_args); in TEST()[all …]
104 uint64_t TORCH_API computeFlops(
577 uint64_t computeFlops( in computeFlops() function
247 evt.setFlops(torch::profiler::impl::computeFlops( in pushRange()
1043 ? torch::profiler::impl::computeFlops(e.name_, e.extra_args_)