Home
last modified time | relevance | path

Searched refs:algorithm_ (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.h65 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;
/external/tensorflow/tensorflow/compiler/xla/service/
Dheap_simulator.cc283 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()
Dheap_simulator.h167 const std::unique_ptr<HeapAlgorithm> algorithm_; variable
276 : algorithm_(std::move(algorithm)) {} in DecreasingSizeRunsHeap()
296 const std::unique_ptr<HeapAlgorithm> algorithm_; variable
Dhlo_memory_scheduler.h108 MemorySchedulerAlgorithm algorithm_; variable
Dhlo_memory_scheduler.cc612 : size_function_(size_function), algorithm_(algorithm) {} in HloMemoryScheduler()
616 ScheduleModule(module, size_function_, algorithm_)); in Run()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h771 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_;
Dblas.cc71 string AlgorithmConfig::ToString() const { return absl::StrCat(algorithm_); } in ToString()
Dblas.h136 AlgorithmType algorithm() const { return algorithm_; } in algorithm()
137 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()
143 AlgorithmType algorithm_ = kDefaultAlgorithm; variable
149 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_;