Lines Matching refs:computeFlops

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()
2889       torch::profiler::impl::computeFlops(std::string("aten::add"), extra_args);  in TEST()
2897       torch::profiler::impl::computeFlops(std::string("aten::mul"), extra_args);  in TEST()