Searched refs:kReduction (Results 1 – 2 of 2) sorted by relevance
395 BiasAddGradGPUConfig() : mode_(BiasAddGradGPUMode::kReduction) {} in BiasAddGradGPUConfig()400 if (mode_ == BiasAddGradGPUMode::kReduction) { in ToString()592 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()604 if (algo_config.get_mode() == BiasAddGradGPUMode::kReduction) { in Compute()
54 kReduction = 2, enumerator