| 161 | } |
| 162 | |
| 163 | void NodeExecStatsWrapper::SetOutput(int slot, const Tensor* tensor) { |
| 164 | DCHECK(tensor); |
| 165 | NodeOutput* node_output = stats_->add_output(); |
| 166 | node_output->set_slot(slot); |
| 167 | tensor->FillDescription(node_output->mutable_tensor_description()); |
| 168 | } |
| 169 | |
| 170 | void NodeExecStatsWrapper::SetReferencedTensors( |
| 171 | const TensorReferenceVector& tensors) { |
no test coverage detected