Searched refs:algorithms_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator.h | 387 : algorithms_(std::move(*algorithms)) {} in ChooseBestHeapAlgorithm()
391 for (auto& algorithm : algorithms_) { in Alloc()
397 for (auto& algorithm : algorithms_) { in Free()
405 std::vector<std::unique_ptr<HeapAlgorithm>> algorithms_;
|
| D | heap_simulator.cc | 927 DCHECK(!algorithms_.empty()); in Finish()
928 std::vector<Result> results(algorithms_.size()); in Finish()
931 for (int i = 0; i < algorithms_.size(); ++i) { in Finish()
932 results[i] = algorithms_[i]->Finish(); in Finish()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 447 ctx, &algorithms_, &algorithms_set_already_); in MatMulOp()
512 ctx, a_float, b_float, dim_pair, &algorithms_, use_autotune_, in Compute()
518 ctx, a, b, dim_pair, &algorithms_, use_autotune_, out); in Compute()
523 std::vector<int64> algorithms_; member in tensorflow::MatMulOp
|