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