Searched refs:executeWithOutputKernel (Results 1 – 1 of 1) sorted by relevance
94 auto executeWithOutputKernel = [&](auto output_kernel) { in operator ()() local118 executeWithOutputKernel(WithBiasAdd<T>(bias_add_args)); in operator ()()121 executeWithOutputKernel(WithBiasAddAndRelu<T>(bias_add_args)); in operator ()()124 executeWithOutputKernel(WithBiasAddAndRelu6<T>(bias_add_args)); in operator ()()127 executeWithOutputKernel(WithBiasAddAndElu<T>(bias_add_args)); in operator ()()130 executeWithOutputKernel(WithBiasAddAndLeakyRelu<T>(bias_add_args)); in operator ()()