| 885 | } |
| 886 | |
| 887 | Status HloCostAnalysis::HandleSort(const HloInstruction* sort) { |
| 888 | // This assumes a comparison based N*log(N) algorithm. As for all ops, the |
| 889 | // actual properties of the op depend on the backend implementation. |
| 890 | int64 elements = ShapeUtil::ElementsIn(sort->operand(0)->shape()); |
| 891 | current_properties_[kFlopsKey] = elements * tensorflow::Log2Ceiling(elements); |
| 892 | return Status::OK(); |
| 893 | } |
| 894 | |
| 895 | Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { |
| 896 | // Since the number of iterations of the while node will not always be |
nothing calls this directly
no test coverage detected