| 251 | } |
| 252 | |
| 253 | int64 ExecStep::cpu_exec_micros() const { |
| 254 | int64 total = 0; |
| 255 | // Normally, an op can only be scheduled on 1 device. |
| 256 | for (const auto& execs : cpu_execs_) { |
| 257 | // An op can be scheduled multiple times in while-loop. |
| 258 | for (const auto& exec : execs.second) { |
| 259 | total += exec.second; |
| 260 | } |
| 261 | } |
| 262 | return total; |
| 263 | } |
| 264 | |
| 265 | std::vector<int64> ShapeProtoToVec(const TensorShapeProto& shape_pb) { |
| 266 | std::vector<int64> shape_vec; |
no outgoing calls
no test coverage detected