Method ResetDefinitionEvent

tensorflow/compiler/jit/xla_tensor.cc:94–99

}

void XlaTensor::ResetDefinitionEvent(std::shared_ptr<se::Event> event,
                                     se::Stream* stream) {
  mutex_lock lock(mu_);
  definition_event_ = std::move(event);
  streams_defined_on_ = {stream};
}

Status XlaTensor::RefreshStatusOfStreams() {
  mutex_lock lock(mu_);