Searched refs:block_epilogue_generator (Results 1 – 2 of 2) sorted by relevance
125 BlockEpilogueGenerator block_epilogue_generator = {})128 block_epilogue_generator_(std::move(block_epilogue_generator)) {} in tile_element_generator_()
3203 const BlockEpilogueGenerator& block_epilogue_generator = in EmitKernel() local3205 if (block_epilogue_generator) { in EmitKernel()3206 block_epilogue_generator(unnested_hlo, kernel_info); in EmitKernel()