Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc1378 RematStrategy best_strategy; in PickRematerializationCandidates() local
1430 best_strategy = strategy; in PickRematerializationCandidates()
1431 best_strategy.compact_shape = compact_shape; in PickRematerializationCandidates()
1482 best_strategy.kind = RematStrategy::kRecompute; in PickRematerializationCandidates()
1499 return {best_items, best_strategy}; in PickRematerializationCandidates()
1757 RematStrategy best_strategy; in RematerializeBestBlock() local
1758 std::tie(best_items, best_strategy) = in RematerializeBestBlock()
1769 if (best_strategy.kind == RematStrategy::kCompress) { in RematerializeBestBlock()
1775 best_items[0], best_strategy.compact_shape)) in RematerializeBestBlock()
1781 best_strategy.compact_shape, instruction_list)); in RematerializeBestBlock()