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;
395 BiasAddGradGPUConfig() : mode_(BiasAddGradGPUMode::kReduction) {} in BiasAddGradGPUConfig()397 if (mode_ == BiasAddGradGPUMode::kNative) { in ToString()400 if (mode_ == BiasAddGradGPUMode::kReduction) { in ToString()405 BiasAddGradGPUMode get_mode() const { return mode_; } in get_mode()406 void set_mode(BiasAddGradGPUMode val) { mode_ = val; } in set_mode()417 BiasAddGradGPUMode mode_;578 best_result.set_algorithm(BiasAddGradGPUMode::kNative); in Compute()592 best_result.set_algorithm(BiasAddGradGPUMode::kReduction); in Compute()604 if (algo_config.get_mode() == BiasAddGradGPUMode::kReduction) { in Compute()