Searched refs:BuildKernelThunk (Results 1 – 2 of 2) sorted by relevance
358 BuildKernelThunk(dot, /*implements_whole_instruction=*/true)); in HandleDot()369 BuildKernelThunk(convolution, /*implements_whole_instruction=*/true)); in HandleConvolution()586 thunks.push_back(BuildKernelThunk( in HandleFusion()607 BuildKernelThunk(fusion, in HandleFusion()652 BuildKernelThunk(fusion, /*implements_whole_instruction=*/true); in HandleFusion()771 BuildKernelThunk(tuple, /*implements_whole_instruction=*/true)); in HandleTuple()796 thunks.push_back(BuildKernelThunk(select_and_scatter, in HandleSelectAndScatter()998 std::unique_ptr<KernelThunk> rng_thunk = BuildKernelThunk( in HandleRng()1015 BuildKernelThunk(rng, /*implements_whole_instruction=*/false); in HandleRng()1047 BuildKernelThunk(scatter, in HandleScatter()[all …]
316 std::unique_ptr<KernelThunk> BuildKernelThunk(