Searched refs:algorithm_ (Results 1 – 8 of 8) sorted by relevance
65 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;
283 algorithm_->AccountForSubcomputationMemory( in RunComputation()356 algorithm_(std::move(algorithm)), in HeapSimulator()389 algorithm_->Alloc(buffer, size); in Alloc()420 algorithm_->Free(buffer, size); in Free()469 Result result = algorithm_->Finish(); in Finish()581 return algorithm_->Finish(); in Finish()606 algorithm_->Alloc(op.buffer, op.size); in CallAndDrainRun()608 algorithm_->Free(op.buffer, op.size); in CallAndDrainRun()
167 const std::unique_ptr<HeapAlgorithm> algorithm_; variable276 : algorithm_(std::move(algorithm)) {} in DecreasingSizeRunsHeap()296 const std::unique_ptr<HeapAlgorithm> algorithm_; variable
108 MemorySchedulerAlgorithm algorithm_; variable
612 : size_function_(size_function), algorithm_(algorithm) {} in HloMemoryScheduler()616 ScheduleModule(module, size_function_, algorithm_)); in Run()
771 return algorithm_.has_value() &&775 AlgorithmDesc algorithm() const { return *algorithm_; }776 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; }785 absl::optional<AlgorithmDesc> algorithm_;801 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {}803 : algorithm_(algorithm), algorithm_no_scratch_(algorithm_no_scratch) {}804 absl::optional<AlgorithmDesc> algorithm() const { return algorithm_; }805 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; }813 return this->algorithm_ == other.algorithm_ &&822 absl::optional<AlgorithmDesc> algorithm_;
71 string AlgorithmConfig::ToString() const { return absl::StrCat(algorithm_); } in ToString()
136 AlgorithmType algorithm() const { return algorithm_; } in algorithm()137 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()143 AlgorithmType algorithm_ = kDefaultAlgorithm; variable149 AlgorithmConfig() : algorithm_(kDefaultAlgorithm) {} in AlgorithmConfig()150 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()151 AlgorithmType algorithm() const { return algorithm_; } in algorithm()152 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()154 return this->algorithm_ == other.algorithm_;162 AlgorithmType algorithm_;