Searched refs:BiasAddGradGPUMode (Results 1 – 2 of 2) sorted by relevance
51 enum class BiasAddGradGPUMode { enum65 return (algorithm_ != BiasAddGradGPUMode::kInvalid && in is_valid()68 BiasAddGradGPUMode algorithm() const { return algorithm_; } in algorithm()69 void set_algorithm(BiasAddGradGPUMode val) { algorithm_ = val; } in set_algorithm()74 BiasAddGradGPUMode algorithm_ = BiasAddGradGPUMode::kInvalid;
363 BiasAddGradGPUConfig() : mode_(BiasAddGradGPUMode::kReduction) {} in BiasAddGradGPUConfig()365 if (mode_ == BiasAddGradGPUMode::kNative) { in ToString()368 if (mode_ == BiasAddGradGPUMode::kReduction) { in ToString()373 BiasAddGradGPUMode get_mode() const { return mode_; } in get_mode()374 void set_mode(BiasAddGradGPUMode val) { mode_ = val; } in set_mode()385 BiasAddGradGPUMode mode_;546 best_result.set_algorithm(BiasAddGradGPUMode::kNative); in Compute()560 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()572 if (algo_config.get_mode() == BiasAddGradGPUMode::kReduction) { in Compute()